(#3) Defense: Add dataset name to tags

betarixm · May 2, 2022 · 4add584 · 4add584
1 parent d798890
commit 4add584
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/defense/models.py b/src/defense/models.py
@@ -169,13 +169,13 @@ def custom_callbacks(self) -> List[keras.callbacks.Callback]:
             y = defense_model.predict(x)
 
             tf.summary.image(
-                f"(Defense) Original images",
+                f"(Defense)[{args.dataset}] Original images",
                 concat_batch_images(x),
                 step=idx,
             )
 
             tf.summary.image(
-                f"(Defense) {args.defense.upper()} processing result",
+                f"(Defense)[{args.dataset}] {args.defense.upper()} processing result",
                 concat_batch_images(y),
                 step=idx,
             )