Skip to content

Commit 790dea2

Browse files
committed
Fixing failing attack test as targets not defined
Signed-off-by: Kieran Fraser <[email protected]>
1 parent 7e46923 commit 790dea2

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

art/attacks/evasion/adversarial_patch/adversarial_patch_pytorch.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -609,7 +609,8 @@ def __getitem__(self, idx):
609609
"scores": target["scores"][idx].to(self.estimator.device),
610610
}
611611
)
612-
_ = self._train_step(images=images, target=targets, mask=None)
612+
target = targets
613+
_ = self._train_step(images=images, target=target, mask=None)
613614
else:
614615
for images, target, mask_i in data_loader:
615616
images = images.to(self.estimator.device)
@@ -625,8 +626,9 @@ def __getitem__(self, idx):
625626
"scores": target["scores"][idx].to(self.estimator.device),
626627
}
627628
)
629+
target = targets
628630
mask_i = mask_i.to(self.estimator.device)
629-
_ = self._train_step(images=images, target=targets, mask=mask_i)
631+
_ = self._train_step(images=images, target=target, mask=mask_i)
630632

631633
# Write summary
632634
if self.summary_writer is not None: # pragma: no cover

0 commit comments

Comments
 (0)