Skip to content

Commit 1966edf

Browse files
authored
Merge pull request #860 from Trusted-AI/development_issue_840
Update AutoProjectedGradientDescent for targeted attacks
2 parents 1deea2b + 73665b9 commit 1966edf

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

art/attacks/evasion/auto_projected_gradient_descent.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -508,7 +508,11 @@ def generate(self, x: np.ndarray, y: Optional[np.ndarray] = None, **kwargs) -> n
508508

509509
f_k_p_1 = self.estimator.loss(x=x_k_p_1, y=y_batch, reduction="mean")
510510

511-
if f_k_p_1 > self.f_max:
511+
if f_k_p_1 == 0.0:
512+
x_k = x_k_p_1.copy()
513+
break
514+
515+
if (not self.targeted and f_k_p_1 > self.f_max) or (self.targeted and f_k_p_1 < self.f_max):
512516
self.count_condition_1 += 1
513517
self.x_max = x_k_p_1
514518
self.x_max_m_1 = x_k

0 commit comments

Comments
 (0)