Update adversarial_regularization.py

Author: wangbingnan136

neural_structured_learning/keras/adversarial_regularization.py

@@ -21,6 +21,7 @@
 import functools
 import types
 from keras.utils import traceback_utils
+from keras.saving import save
 
 import attr
 import neural_structured_learning.configs as nsl_configs
@@ -695,11 +696,6 @@ def call(self, inputs, **kwargs):
         scaled_adv_loss, name='scaled_adversarial_loss', aggregation='mean')
     return outputs
 
-  def save(self, *args, **kwargs):
-    raise NotImplementedError(
-        'Saving `AdversarialRegularization` models is currently not supported. '
-        'Consider using `save_weights` or saving the `base_model`.')
-
 
   @traceback_utils.filter_traceback
   def save(self,
@@ -753,8 +749,6 @@ def save(self,
 
 
 
-
-
   def perturb_on_batch(self, x, **config_kwargs):
     """Perturbs the given input to generates adversarial examples.