import tensorflow as tf

                            'num of residual units')
tf.app.flags.DEFINE_string('Optimizer', 'mom',
                           'The optimizer used to train the model.')
tf.app.flags.DEFINE_bool('RCE_train', False,
                         'Whether use RCE to train the model.')
tf.app.flags.DEFINE_string('attack_method', 'fgsm',
