Update prototype.py

AISafety-HKUST · Oct 24, 2024 · a3b0327 · a3b0327
1 parent 6e598c1
commit a3b0327
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/attack/prototype.py b/attack/prototype.py
@@ -147,7 +147,7 @@ def benign_prepare(self):
         train_label_transform, \
         test_dataset_without_transform, \
         test_img_transform, \
-        test_label_transform,_ = dataset_and_transform_generate(args)
+        test_label_transform = dataset_and_transform_generate(args)