Skip to content

Commit 87fcd6e

Browse files
committed
Allow to skip stage 2 (i.e. max_iter_2=0)
Running the imperceptible attack without stage 2 equals the Carlini and Wagner audio attack (see paper link: https://arxiv.org/abs/1801.01944). Signed-off-by: Hessel Tuinhof <[email protected]>
1 parent 43890d1 commit 87fcd6e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

art/attacks/evasion/imperceptible_asr/imperceptible_asr.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -494,8 +494,8 @@ def _check_params(self) -> None:
494494

495495
if not isinstance(self.max_iter_2, int):
496496
raise ValueError("The maximum number of iterations for stage 2 must be of type int.")
497-
if self.max_iter_2 <= 0:
498-
raise ValueError("The maximum number of iterations for stage 2 must be greater than 0.")
497+
if self.max_iter_2 < 0:
498+
raise ValueError("The maximum number of iterations for stage 2 must be non-negative.")
499499

500500
if not isinstance(self.learning_rate_1, float):
501501
raise ValueError("The learning rate for stage 1 must be of type float.")

0 commit comments

Comments
 (0)