Add simple baseline for LFNA

2021-04-29 16:30:47 +08:00
parent 2c56938ee7
commit 14905d0011
8 changed files with 296 additions and 307 deletions
--- a/lib/log_utils/logger.py
+++ b/lib/log_utils/logger.py
@@ -59,8 +59,10 @@ class Logger(object):
        )

    def path(self, mode):
-        valids = ("model", "best", "info", "log")
-        if mode == "model":
+        valids = ("model", "best", "info", "log", None)
+        if mode is None:
+            return self.log_dir
+        elif mode == "model":
            return self.model_dir / "seed-{:}-basic.pth".format(self.seed)
        elif mode == "best":
            return self.model_dir / "seed-{:}-best.pth".format(self.seed)