| """Explicit V4 training entry; importing it never starts training.""" | |
| import sys | |
| import argparse | |
| import tomllib | |
| from pathlib import Path | |
| ROOT = Path(__file__).resolve().parents[2] | |
| sys.path.insert(0, str(ROOT / "src")) | |
| from vimeml.training.model_factory_v4 import register_v4 | |
| from vimeml.training.train import main as training_main | |
| def main(argv=None): | |
| parser = argparse.ArgumentParser(description=__doc__) | |
| parser.add_argument("--config", type=Path, default=ROOT / "configs/train-v4-b.toml") | |
| parser.add_argument("--resume", action="store_true") | |
| parser.add_argument("--dry-run", action="store_true") | |
| args = parser.parse_args(argv) | |
| config = tomllib.loads(args.config.read_text(encoding="utf-8")) | |
| if config.get("architecture") != "tiny_gpt_v4": | |
| parser.error("This entry point accepts only tiny_gpt_v4 configurations") | |
| register_v4() | |
| # Reuse the established tracking dispatch, including its offline mode. | |
| forwarded = ["--config", str(args.config.resolve())] | |
| if args.resume: | |
| forwarded.append("--resume") | |
| if args.dry_run: | |
| forwarded.append("--dry-run") | |
| return training_main(forwarded) | |
| if __name__ == "__main__": | |
| main() | |