Skip to content

Commit 4dd03fb

Browse files
author
Beat Buesser
committed
Make unit test more robust
Signed-off-by: Beat Buesser <[email protected]>
1 parent 56f9c9c commit 4dd03fb

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/defences/trainer/test_adversarial_trainer_FBF.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def _get_adv_trainer():
3434
trainer = None
3535
if framework == "pytorch":
3636
classifier, _ = image_dl_estimator()
37-
trainer = AdversarialTrainerFBFPyTorch(classifier, eps=0.1)
37+
trainer = AdversarialTrainerFBFPyTorch(classifier, eps=0.05)
3838
if framework == "scikitlearn":
3939
trainer = None
4040

@@ -75,7 +75,7 @@ def test_adversarial_trainer_fbf_pytorch_fit_and_predict(get_adv_trainer, fix_ge
7575
)
7676

7777
assert accuracy == 0.32
78-
assert accuracy_new == 0.58
78+
assert accuracy_new == 0.63
7979

8080
trainer.fit(x_train_mnist, y_train_mnist, nb_epochs=20, validation_data=(x_train_mnist, y_train_mnist))
8181

0 commit comments

Comments
 (0)