Merge pull request #897 from industrysc/sc

Import path correction for Madry's CIFAR-10 model
cleverhans-lab · Nov 16, 2018 · e204014 · e204014
2 parents a562394 + c45fa04
commit e204014
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/madry_lab_challenges/cifar10/attack_model.py b/examples/madry_lab_challenges/cifar10/attack_model.py
@@ -44,8 +44,8 @@ def main(argv):
     x = tf.placeholder(tf.float32, shape=(None, 32, 32, 3))
     y = tf.placeholder(tf.float32, shape=(None, 10))
 
-    from madry_cifar10_model import make_madry_wresnet
-    model = make_madry_wresnet()
+    from cleverhans.model_zoo.madry_lab_challenges.cifar10_model import make_wresnet
+    model = make_wresnet()
 
     saver = tf.train.Saver()