add from_logits=True to begin addressing issue #2227

Signed-off-by: GiulioZizzo <[email protected]>

Author: GiulioZizzo

tests/defences/trainer/test_adversarial_trainer_trades_pytorch.py

@@ -34,7 +34,7 @@ def _get_adv_trainer():
         if framework in ["tensorflow", "tensorflow2v1"]:
             trainer = None
         if framework == "pytorch":
-            classifier, _ = image_dl_estimator()
+            classifier, _ = image_dl_estimator(from_logits=True)
             attack = ProjectedGradientDescent(
                 classifier,
                 norm=np.inf,
@@ -121,7 +121,6 @@ def test_adversarial_trainer_trades_pytorch_fit_generator_and_predict(
         assert y_test_mnist.shape[-1] == 10
     if label_format == "numerical":
         y_test_mnist = np.argmax(y_test_mnist, axis=1)
-        y_train_mnist = np.argmax(y_train_mnist, axis=1)
 
     generator = image_data_generator()