Skip to content

Commit de5134b

Browse files
authored
Merge pull request #1758 from Trusted-AI/development_issue_1644
Update default value for nb_epochs in AdversarialTrainerMadryPGD
2 parents 8aa0aa6 + 4b13c60 commit de5134b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

art/defences/trainer/adversarial_trainer_madry_pgd.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ class AdversarialTrainerMadryPGD(Trainer):
5454
def __init__(
5555
self,
5656
classifier: "CLASSIFIER_LOSS_GRADIENTS_TYPE",
57-
nb_epochs: Optional[int] = 391,
57+
nb_epochs: Optional[int] = 205,
5858
batch_size: Optional[int] = 128,
5959
eps: Union[int, float] = 8,
6060
eps_step: Union[int, float] = 2,

0 commit comments

Comments
 (0)