We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 8aa0aa6 + 4b13c60 commit de5134bCopy full SHA for de5134b
art/defences/trainer/adversarial_trainer_madry_pgd.py
@@ -54,7 +54,7 @@ class AdversarialTrainerMadryPGD(Trainer):
54
def __init__(
55
self,
56
classifier: "CLASSIFIER_LOSS_GRADIENTS_TYPE",
57
- nb_epochs: Optional[int] = 391,
+ nb_epochs: Optional[int] = 205,
58
batch_size: Optional[int] = 128,
59
eps: Union[int, float] = 8,
60
eps_step: Union[int, float] = 2,
0 commit comments