Update mnist_tutorial_jsma.py
@@ -20,7 +20,8 @@ from cleverhans.dataset import MNIST
from cleverhans.loss import CrossEntropy
from cleverhans.utils import other_classes, set_log_level
from cleverhans.utils import pair_visual, grid_visual, AccuracyReport
-from cleverhans.utils_tf import train, model_eval, model_argmax
+from cleverhans.utils_tf import model_eval, model_argmax
+from cleverhans.train import train
from cleverhans_tutorials.tutorial_models import ModelBasicCNN
FLAGS = flags.FLAGS
@@ -97,8 +98,7 @@ def mnist_tutorial_jsma(train_start=0, train_end=60000, test_start=0,
}
sess.run(tf.global_variables_initializer())
rng = np.random.RandomState([2017, 8, 30])
- train(sess, loss, x, y, x_train, y_train, args=train_params,
- rng=rng)
+ train(sess, loss, x_train, y_train, args=train_params, rng=rng)
# Evaluate the accuracy of the MNIST model on legitimate test examples
eval_params = {'batch_size': batch_size}
