(#7) Attack: Logging defense method name

betarixm · May 2, 2022 · 8a1fe48 · 8a1fe48
1 parent deb2928
commit 8a1fe48
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/attack/attack.py b/src/attack/attack.py
@@ -92,5 +92,5 @@
     print(f"    - Normal:       {acc.result()}")
     print(f"    - Under Attack: {acc_under_attack.result()}")
 
-    if args.defense:
-        print(f"    - With Defense: {acc_with_defense.result()}")
+    if defense_model is not None:
+        print(f"    - With {args.defense.upper()}: {acc_with_defense.result()}")