Make all_convolutional picklable
@@ -5,13 +5,13 @@ import math
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
