Open timchenggu123 opened 2 years ago
on this line https://github.com/DSE-MSU/DeepRobust/blob/master/deeprobust/image/defense/pgdtraining.py#L219
adversary = PGD(self.model)
You guys forgot to pass in the self.device parameter, which results in inconsistency between the device in the PGD class and the PGDTraining class. I believe the correct code should be
adversary = PGD(self.model, device=self.device)
Can you guys fix that code? Thanks! :D
find the same error
GumpCode
commented
2 years ago
on this line https://github.com/DSE-MSU/DeepRobust/blob/master/deeprobust/image/defense/pgdtraining.py#L219
adversary = PGD(self.model)You guys forgot to pass in the self.device parameter, which results in inconsistency between the device in the PGD class and the PGDTraining class. I believe the correct code should be
adversary = PGD(self.model, device=self.device)Can you guys fix that code? Thanks! :D