Генерация изображений соперников FGSM с использованием библиотеки cleverhans

Question

Я использую архитектуру L eNet 5 в наборе данных MNIST для получения данных о состязаниях с использованием библиотеки cleverhans. Но во время генерации изображения выдает ошибку при компиляции этих строк

x = tf.convert_to_tensor(x_train[0])
adv_x = fgsm.generate(x, eps = 0.3)

шаги должны быть длиной 1, 1 или 3, но были 2

Модель I Написано ниже:

model = keras.Sequential([

    keras.layers.Conv2D(filters = 6, kernel_size = (5,5), strides = 1, 
                        activation = 'tanh', input_shape = (28,28,1), padding = "same"),

    keras.layers.AveragePooling2D(pool_size=(2, 2), strides= 1 , padding='valid'),

    keras.layers.Conv2D(filters = 16, kernel_size = (5,5) , strides = 1,
                        activation = 'tanh', padding = "valid"),
    keras.layers.AveragePooling2D(pool_size = (2,2), strides = 1, padding = 'valid'),

    keras.layers.Conv2D(filters = 120, kernel_size=(5,5), strides = 1,
                        activation='tanh', padding="valid"),

    keras.layers.Flatten(),
    keras.layers.Dense(84, activation ='tanh'),
    keras.layers.Dense(10, activation= 'softmax')
])
model.compile(loss= keras.losses.categorical_crossentropy, optimizer='SGD', metrics = ['accuracy'])

Итак, как вы можете видеть, я использую шаги длины 1, но все еще получаю эту ошибку.