Skip to content

Commit d3280eb

Browse files
authored
Merge pull request #2558 from Trusted-AI/development_issue_2553
Fix missing transfer to device in ProjectedGradientDescentPyTorch
2 parents fd63952 + d794c1f commit d3280eb

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

art/attacks/evasion/projected_gradient_descent/projected_gradient_descent_pytorch.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -497,7 +497,10 @@ def _projection(
497497
if (suboptimal or norm == 2) and norm != np.inf: # Simple rescaling
498498
values_norm = torch.linalg.norm(values_tmp, ord=norm, dim=1, keepdim=True) # (n_samples, 1)
499499
values_tmp = values_tmp * values_norm.where(
500-
values_norm == 0, torch.minimum(torch.ones(1), torch.tensor(eps).to(values_tmp.device) / values_norm)
500+
values_norm == 0,
501+
torch.minimum(
502+
torch.ones(1).to(values_tmp.device), torch.tensor(eps).to(values_tmp.device) / values_norm
503+
),
501504
)
502505
else: # Optimal
503506
if norm == np.inf: # Easy exact case

0 commit comments

Comments
 (0)