File size: 612 Bytes
e52b09c |
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'))
|