Merge pull request #989 from anant90/master

Make all_convolutional picklable
cleverhans-lab · Feb 21, 2019 · 36c5e32 · 36c5e32
2 parents 16902ce + fb804af
commit 36c5e32
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/cleverhans/model_zoo/all_convolutional.py b/cleverhans/model_zoo/all_convolutional.py
@@ -5,13 +5,13 @@
 import tensorflow as tf
 
 from cleverhans import initializers
-from cleverhans.model import Model
+from cleverhans.serial import NoRefModel
 
 
-class ModelAllConvolutional(Model):
+class ModelAllConvolutional(NoRefModel):
   def __init__(self, scope, nb_classes, nb_filters, input_shape, **kwargs):
     del kwargs
-    Model.__init__(self, scope, nb_classes, locals())
+    NoRefModel.__init__(self, scope, nb_classes, locals())
     self.nb_filters = nb_filters
     self.input_shape = input_shape