code

viraweb123 · Aug 14, 2024 · a8d7d82 · a8d7d82
1 parent 377c898
commit a8d7d82
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train/code/main.py b/train/code/main.py
@@ -11,9 +11,9 @@
 log_base_dir = "/train/log"        
 input_base_dir = "/train/input"    
 
-output_dir = os.path.join(output_base_dir, "code_log",log_dir_name)
+output_dir = os.path.join(output_base_dir, "trained_model",log_dir_name)
 logging_dir = os.path.join(log_base_dir, "model_log", log_dir_name)
-log_dir = os.path.join(log_base_dir, log_dir_name)
+log_dir = os.path.join(log_base_dir, "code_log",log_dir_name)
 
 os.makedirs(output_dir, exist_ok=True)
 os.makedirs(log_dir, exist_ok=True)