Merge pull request #2461 from Trusted-AI/dev_1.18.1

Update to ART 1.18.1
Trusted-AI · Jul 3, 2024 · 4508b6e · 4508b6e
2 parents a20c78f + 7044477
commit 4508b6e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/art/attacks/evasion/projected_gradient_descent/projected_gradient_descent_pytorch.py b/art/attacks/evasion/projected_gradient_descent/projected_gradient_descent_pytorch.py
@@ -497,11 +497,13 @@ def _projection(
         if (suboptimal or norm == 2) and norm != np.inf:  # Simple rescaling
             values_norm = torch.linalg.norm(values_tmp, ord=norm, dim=1, keepdim=True)  # (n_samples, 1)
             values_tmp = values_tmp * values_norm.where(
-                values_norm == 0, torch.minimum(torch.ones(1), torch.Tensor(eps) / values_norm)
+                values_norm == 0, torch.minimum(torch.ones(1), torch.tensor(eps).to(values_tmp.device) / values_norm)
             )
         else:  # Optimal
             if norm == np.inf:  # Easy exact case
-                values_tmp = values_tmp.sign() * torch.minimum(values_tmp.abs(), torch.Tensor(eps))
+                values_tmp = values_tmp.sign() * torch.minimum(
+                    values_tmp.abs(), torch.tensor(eps).to(values_tmp.device)
+                )
             elif norm >= 1:  # Convex optim
                 raise NotImplementedError(
                     "Finite values of `norm_p >= 1` are currently not supported with `suboptimal=False`."