claude_refactor_v3: New train_gtauav.py, added entry point main.py, added utils

src/main.py

@@ -0,0 +1,54 @@
+"""Entry point: load configs and run training.
+
+Usage:
+    python -m src.main gtauav_balanced
+    python -m src.main gtauav_balanced_sofia_v1
+"""
+
+from __future__ import annotations
+
+import logging
+import sys
+
+import coloredlogs
+
+from src.conf.config_loader import load_all_configs
+from src.training.train_gtauav_old import train
+from src.utils.path_utils import get_proj_dir
+
+logger = logging.getLogger("caption_test")
+
+
+def main() -> None:
+    coloredlogs.install(
+        level="INFO",
+        logger=logger,
+        fmt="%(asctime)s %(name)s %(levelname)s %(message)s",
+    )
+
+    if len(sys.argv) != 2:
+        raise SystemExit(
+            "Usage: python -m src.main <preset_name>\n"
+            "Example: python -m src.main gtauav_balanced\n"
+            "  available presets are subdirectories under in/config_files/",
+        )
+    preset_name = sys.argv[1]
+
+    proj_dir = get_proj_dir()
+    path2cfg = f"{proj_dir}in/config_files/"           # per REQUIREMENTS_GIN_STYLE.md §5
+
+    configs = load_all_configs(path2cfg, preset_name)
+
+    train(
+        pipeline_cfg=configs["pipeline"],
+        hardware_cfg=configs["hardware"],
+        training_cfg=configs["training"],
+        tracking_cfg=configs["tracking"],
+        models_common_cfg=configs["models_common"],
+        models_cfg=configs["models"],
+    )
+
+
+if __name__ == "__main__":
+    main()
+