File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed
art/attacks/evasion/adversarial_patch Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments