Refactor run directory creation to use a unique naming function in training scripts

train_all_future.py

@@ -15,7 +15,6 @@ import json
 import logging
 import math
 import time
-from datetime import datetime
 from pathlib import Path
 from typing import Any, Dict
 
@@ -32,6 +31,7 @@ from models import DeepHealth
 from targets import CHECKUP_IDX, PAD_IDX
 from train_util import (
     configure_torch_for_training,
+    create_unique_run_dir,
     load_extra_info_types_file,
     resolve_device,
     save_checkpoint,
@@ -295,10 +295,9 @@ def main() -> None:
     device = resolve_device(args.device)
     configure_torch_for_training(device)
 
-    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-    run_name = f"{args.time_mode}_{args.dist_mode}_all_future_pure_disease_{timestamp}"
-    run_dir = Path("runs") / run_name
-    run_dir.mkdir(parents=True, exist_ok=False)
+    run_dir, run_name = create_unique_run_dir(
+        lambda timestamp: f"{args.time_mode}_{args.dist_mode}_all_future_pure_disease_{timestamp}"
+    )
     logger = setup_logging(run_dir)
 
     logger.info(f"Starting all-future training run: {run_name}")