Move CWL2 to CW as Adversarian suggestion

Harry24k · Nov 29, 2023 · 6ba76d4 · 6ba76d4
1 parent a464809
commit 6ba76d4
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/torchattacks/__init__.py b/torchattacks/__init__.py
@@ -27,11 +27,11 @@
 from .attacks.pifgsmpp import PIFGSMPP
 
 # L2 attacks
+from .attacks.cw import CW
 from .attacks.cwl0 import CWL0
-from .attacks.cwl2 import CWL2
 from .attacks.cwlinf import CWLinf
+from .attacks.cwbs import CWBS
 from .attacks.cwbsl0 import CWBSL0
-from .attacks.cwbsl2 import CWBSL2
 from .attacks.cwbslinf import CWBSLinf
 from .attacks.pgdl2 import PGDL2
 from .attacks.pgdrsl2 import PGDRSL2
@@ -85,11 +85,11 @@
     "EADEN",
     "PIFGSM",
     "PIFGSMPP",
+    "CW",
     "CWL0",
-    "CWL2",
     "CWLinf",
+    "CWBS",
     "CWBSL0",
-    "CWBSL2",
     "CWBSLinf",
     "PGDL2",
     "DeepFool",

diff --git a/torchattacks/attacks/cwl2.py → torchattacks/attacks/cw.py b/torchattacks/attacks/cwl2.py → torchattacks/attacks/cw.py
@@ -5,7 +5,7 @@
 from ..attack import Attack
 
 
-class CWL2(Attack):
+class CW(Attack):
     r"""
     CW in the paper 'Towards Evaluating the Robustness of Neural Networks'
     [https://arxiv.org/abs/1608.04644]

diff --git a/torchattacks/attacks/cwbsl2.py → torchattacks/attacks/cwbs.py b/torchattacks/attacks/cwbsl2.py → torchattacks/attacks/cwbs.py
@@ -5,7 +5,7 @@
 from ..attack import Attack
 
 
-class CWBSL2(Attack):
+class CWBS(Attack):
     r"""
     CW (binary search version) in the paper 'Towards Evaluating the Robustness of Neural Networks'
     [https://arxiv.org/abs/1608.04644]