feat(ai): add checkpoint path as argument on launch

2026-03-22 00:36:20 +00:00 · 2024-01-05 16:34:26 +02:00
parent e6cef45d19
commit 0e5e502898
4 changed files with 14 additions and 11 deletions
--- a/main.py
+++ b/main.py
@@ -32,9 +32,8 @@ group.add_argument(
 parser.add_argument(
    "-t",
    "--train",
-    type=pos_int,
-    nargs=2,
-    metavar=("generations", "parallels"),
+    nargs=3,
+    metavar=("n generations", "n parallels", "checkpoint"),
    help="Trains the AI",
 )