claude_refactor_v3: Updated main (entry point), trainer_new (last version of train_gtauav), check: is extracted evluate() from train to evaluator.py correct in new context

src/main.py

@@ -2,7 +2,7 @@
 
 Usage:
     python -m src.main gtauav_balanced
-    python -m src.main gtauav_balanced_sofia_v1
+    python -m src.main gtauav_balanced_stripnet
 """
 
 from __future__ import annotations
@@ -13,7 +13,7 @@ import sys
 import coloredlogs
 
 from src.conf.config_loader import load_all_configs
-from src.training.train_gtauav_old import train
+from src.training.trainer_new import Trainer
 from src.utils.path_utils import get_proj_dir
 
 logger = logging.getLogger("caption_test")
@@ -39,7 +39,7 @@ def main() -> None:
 
     configs = load_all_configs(path2cfg, preset_name)
 
-    train(
+    trainer = Trainer(
         pipeline_cfg=configs["pipeline"],
         hardware_cfg=configs["hardware"],
         training_cfg=configs["training"],
@@ -48,7 +48,10 @@ def main() -> None:
         models_cfg=configs["models"],
     )
 
+    trainer.train()
+
 
 if __name__ == "__main__":
     main()
 
+