Merge pull request #895 from marrlab/miro_no_dump

fix disk: out of space miro no dump
marrlab · Nov 21, 2024 · 6da7252 · 6da7252
2 parents fce50f5 + cdb7787
commit 6da7252
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/tests/test_miro.py b/tests/test_miro.py
@@ -9,5 +9,5 @@ def test_miro():
     train with MIRO
     """
     args = "--te_d=caltech --task=mini_vlcs --debug --bs=2 --model=erm \
-        --trainer=miro --nname=alexnet"
+        --trainer=miro --nname=alexnet --no_dump"
     utils_test_algo(args)
diff --git a/tests/test_miro2.py b/tests/test_miro2.py
@@ -10,5 +10,5 @@ def test_miro2():
     """
     args = "--te_d=2 --tr_d 0 1 --task=mnistcolor10 --debug --bs=100 --model=erm \
         --trainer=miro --nname=conv_bn_pool_2 \
-        --layers2extract_feats _net_invar_feat.conv_net.5"
+        --layers2extract_feats _net_invar_feat.conv_net.5 --no_dump"
     utils_test_algo(args)
diff --git a/tests/test_miro3.py b/tests/test_miro3.py
@@ -11,7 +11,7 @@ def test_miro3():
     """
     args = "--te_d=caltech --task=mini_vlcs --debug --bs=2 --model=erm \
         --trainer=miro --nname=alexnet \
-        --layers2extract_feats features"
+        --layers2extract_feats features --no_dump"
     with pytest.raises(RuntimeError):
         utils_test_algo(args)
         raise RuntimeError("This is a runtime error")