Update main.py

illidanlab · Mar 4, 2024 · 8376e9b · 8376e9b
1 parent a717632
commit 8376e9b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -314,7 +314,7 @@ def main():
                         help='cutmix probability')
     #weight perturbation
     parser.add_argument('--filter', default=None, type=str,
-                        help='filter with adv or entropy.')
+                        help='AWP if we want to adopt weight peturbation during watermark injection.')
     parser.add_argument('--select_portion', default=0, type=float,
                         help='choose top select_portion samples according to adv/entropy loss')