We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent bc6588d commit 278b005Copy full SHA for 278b005
art/defences/trainer/adversarial_trainer_awp_pytorch.py
@@ -416,9 +416,9 @@ def _weight_perturbation(
416
"Incorrect mode provided for base adversarial training. 'mode' must be among 'PGD' and 'TRADES'."
417
)
418
419
- self._proxy_classifier._optimizer.zero_grad() # pylint: disable=W0212 # type: ignore
+ self._proxy_classifier._optimizer.zero_grad() # type: ignore # pylint: disable=W0212
420
loss.backward()
421
- self._proxy_classifier._optimizer.step() # pylint: disable=W0212 # type: ignore
+ self._proxy_classifier._optimizer.step() # type: ignore # pylint: disable=W0212
422
423
params_dict_proxy, _ = self._calculate_model_params(self._proxy_classifier)
424
0 commit comments