From f1369e13c6f9b8dcee6c62ffc1f4e306c913299e Mon Sep 17 00:00:00 2001 From: Yandi <yandirzm@gmail.com> Date: Mon, 6 Feb 2023 21:08:58 +0100 Subject: [PATCH] refactoring code --- main.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index 76645eb..d0d75d9 100644 --- a/main.py +++ b/main.py @@ -121,6 +121,7 @@ def train(args, cfg): if best_val_loss != None: if val_loss < best_val_loss : network_checkpoint.update(val_loss) + best_val_loss = val_loss scheduler.step(val_loss) @@ -207,6 +208,6 @@ if __name__ == "__main__": config_file = open("config.yml") cfg = yaml.load(config_file, Loader=yaml.FullLoader) - + eval(f"{args.command}(args)") -- GitLab