55 lines
1.3 KiB
Python
55 lines
1.3 KiB
Python
"""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()
|
|
|