File size: 612 Bytes
4cc8ccf
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
diff --git a/train_tuneavideo.py b/train_tuneavideo.py
index 66d51b2..86b2a5d 100644
--- a/train_tuneavideo.py
+++ b/train_tuneavideo.py
@@ -94,8 +94,8 @@ def main(
 
     # Handle the output folder creation
     if accelerator.is_main_process:
-        now = datetime.datetime.now().strftime("%Y-%m-%dT%H-%M-%S")
-        output_dir = os.path.join(output_dir, now)
+        #now = datetime.datetime.now().strftime("%Y-%m-%dT%H-%M-%S")
+        #output_dir = os.path.join(output_dir, now)
         os.makedirs(output_dir, exist_ok=True)
         OmegaConf.save(config, os.path.join(output_dir, 'config.yaml'))