From 78cb661e6ce5521fff3a2c4595fff9c597f97693 Mon Sep 17 00:00:00 2001
From: linyiLYi <48440925+linyiLYi@users.noreply.github.com>
Date: Thu, 27 Apr 2023 09:56:45 +0800
Subject: [PATCH] Revise save_dir

---
 main/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/main/train.py b/main/train.py
index 3a36e57..65606b5 100644
--- a/main/train.py
+++ b/main/train.py
@@ -86,7 +86,7 @@ def main():
     )
 
     # Set the save directory
-    save_dir = "trained_models_test"
+    save_dir = "trained_models"
     os.makedirs(save_dir, exist_ok=True)
 
     # Load the model from file