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