Update test_irm.py

marrlab · Dec 10, 2024 · ab19a3e · ab19a3e
1 parent 78aeafd
commit ab19a3e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_irm.py b/tests/test_irm.py
@@ -17,7 +17,7 @@ def test_irm_sepdom():
     train with Invariant Risk Minimization
     """
     args = "--te_d=caltech --task=mini_vlcs --debug --bs=2 --model=erm \
-        --trainer=irmsepdom --nname=alexnet"
+        --trainer=irmsepdom --nname=alexnet --no_dump"
     utils_test_algo(args)