rename tfe classes
@@ -32,8 +32,8 @@ except AttributeError:
from cleverhans.utils_mnist import data_mnist
from cleverhans.utils import AccuracyReport, set_log_level
from cleverhans.utils_tfe import train, model_eval
-from cleverhans.attacks_tfe import BasicIterativeMethodTFE
-from cleverhans.attacks_tfe import FastGradientMethodTFE
+from cleverhans.attacks_tfe import BasicIterativeMethod
+from cleverhans.attacks_tfe import FastGradientMethod
from cleverhans_tutorials.tutorial_models_tfe import ModelBasicCNNTFE
if tf.executing_eagerly() is True:
@@ -48,10 +48,11 @@ FLAGS = flags.FLAGS
# Keeps track of implemented attacks.
# Maps attack string taken from bash to attack class
# -- 'fgsm' : FastGradientMethod
+# -- 'bim': BasicIterativeMethod
AVAILABLE_ATTACKS = {
- 'fgsm': FastGradientMethodTFE,
- 'bim': BasicIterativeMethodTFE
+ 'fgsm': FastGradientMethod,
+ 'bim': BasicIterativeMethod
}
