code

viraweb123 · Aug 14, 2024 · 698b373 · 698b373
1 parent bfba275
commit 698b373
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train/code/main.py b/train/code/main.py
@@ -12,7 +12,7 @@
 input_base_dir = "/train/input"    
 
 output_dir = os.path.join(output_base_dir, log_dir_name, "trained_model")
-logging_dir = os.path.join(log_base_dir, "model_log")
+logging_dir = os.path.join(log_base_dir, "model_log", log_dir_name)
 log_dir = os.path.join(log_base_dir, log_dir_name)
 
 os.makedirs(output_dir, exist_ok=True)