Update test_observer.py

marrlab · Oct 8, 2023 · ed209dd · ed209dd
1 parent 4deb022
commit ed209dd
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/tests/test_observer.py b/tests/test_observer.py
@@ -26,3 +26,23 @@ def test_deepall():
     del exp
     torch.cuda.empty_cache()
     gc.collect()
+
+
+def test_deepall_trloss():
+    """
+    unit deep all
+    """
+    parser = mk_parser_main()
+    margs = parser.parse_args(["--te_d", "caltech",
+                               "--task", "mini_vlcs",
+                               "--aname", "deepall", "--bs", "2",
+                               "--nname", "conv_bn_pool_2",
+                               "--msel", "loss_tr"
+                               ])
+    exp = Exp(margs)
+    exp.trainer.before_tr()
+    exp.trainer.tr_epoch(0)
+    exp.trainer.observer.update(True)
+    del exp
+    torch.cuda.empty_cache()
+    gc.collect()