PGD torch device inconsistency

[timchenggu123]

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

[GumpCode]

find the same error