Generator runner improvement

2026-04-30 03:21:49 +01:00
parent bb3dfb92d5
commit d75272cf84
4 changed files with 16 additions and 22 deletions
@@ -37,7 +37,7 @@ def main(config_path, *, data_dir_override=None, output_root="generator/outputs"

    cfg = load_config(config_path)

-    run_name    = cfg["run_name"]
+    run_name    = cfg.get("run_name", Path(config_path).stem)
    device      = "cuda" if torch.cuda.is_available() else "cpu"
    data_dir    = data_dir_override or cfg.get("data_dir", "data")
    output_root = Path(output_root)