diff --git a/tests/test_fbopt_irm.py b/tests/test_fbopt_irm.py index fb1f109f6..10cfbcb53 100644 --- a/tests/test_fbopt_irm.py +++ b/tests/test_fbopt_irm.py @@ -10,5 +10,5 @@ def test_mhof_irm(): """ args = "--te_d=0 --task=mnistcolor10 --model=erm \ --trainer=fbopt_irm --nname=conv_bn_pool_2 \ - --k_i_gain_ratio=0.5" + --k_i_gain_ratio=0.5 --no_dump" utils_test_algo(args) diff --git a/tests/test_fbopt_setpoint_rewind.py b/tests/test_fbopt_setpoint_rewind.py index 3ce13079b..3fcc8660e 100644 --- a/tests/test_fbopt_setpoint_rewind.py +++ b/tests/test_fbopt_setpoint_rewind.py @@ -8,5 +8,5 @@ def test_jigen_fbopt(): """ jigen """ - args = "--te_d=caltech --task=mini_vlcs --debug --bs=2 --model=jigen --trainer=fbopt --nname=alexnet --epos=300 --setpoint_rewind" + args = "--te_d=caltech --task=mini_vlcs --debug --bs=2 --model=jigen --trainer=fbopt --nname=alexnet --epos=300 --setpoint_rewind --no_dump" utils_test_algo(args)