Add --force

Author: JasonGross

gbmi/exp_argmax_of_n/run_train_argmax_of_10.py

@@ -11,11 +11,17 @@
     default=",".join(sorted(map(str, SEEDS))),
     help="Comma-separated list of seeds to use",
 )
+parser.add_argument(
+    "--force",
+    choices=["train", "load", "none"],
+    default="train",
+    help="Force training or loading",
+)
 args = parser.parse_args()
 
 for d_vocab in tqdm((64, 128), desc="d_vocab"):
     with tqdm(map(int, args.seeds.split(",")), desc="Seed", leave=False) as pbar:
         for seed in pbar:
             cfg = ARGMAX_OF_10_CONFIG(seed, d_vocab=d_vocab, deterministic=False)
             pbar.set_postfix({"seed": seed, "d_vocab": d_vocab, "cfg": cfg})
-            runtime, model = train_or_load_model(cfg)  # , force="train"
+            runtime, model = train_or_load_model(cfg, force=args.force)

gbmi/exp_argmax_of_n/run_train_argmax_of_20.py

@@ -11,11 +11,17 @@
     default=",".join(sorted(map(str, SEEDS))),
     help="Comma-separated list of seeds to use",
 )
+parser.add_argument(
+    "--force",
+    choices=["train", "load", "none"],
+    default="train",
+    help="Force training or loading",
+)
 args = parser.parse_args()
 
 for d_vocab in tqdm((64, 512), desc="d_vocab"):
     with tqdm(map(int, args.seeds.split(",")), desc="Seed", leave=False) as pbar:
         for seed in pbar:
             cfg = ARGMAX_OF_20_CONFIG(seed, d_vocab=d_vocab, deterministic=False)
             pbar.set_postfix({"seed": seed, "d_vocab": d_vocab, "cfg": cfg})
-            runtime, model = train_or_load_model(cfg)  # , force="train"
+            runtime, model = train_or_load_model(cfg, force=args.force)

gbmi/exp_argmax_of_n/run_train_argmax_of_4.py

@@ -11,11 +11,15 @@
     default=",".join(sorted(map(str, SEEDS))),
     help="Comma-separated list of seeds to use",
 )
+parser.add_argument(
+    "--force",
+    choices=["train", "load", "none"],
+    default="train",
+    help="Force training or loading",
+)
 args = parser.parse_args()
 
 with tqdm(map(int, args.seeds.split(",")), desc="Seed") as pbar:
     for seed in pbar:
         pbar.set_postfix({"seed": seed})
-        runtime, model = train_or_load_model(
-            ARGMAX_OF_4_CONFIG(seed)
-        )  # , force="train"
+        runtime, model = train_or_load_model(ARGMAX_OF_4_CONFIG(seed), force=args.force)

gbmi/exp_argmax_of_n/run_train_argmax_of_5.py

@@ -11,10 +11,16 @@
     default=",".join(sorted(map(str, SEEDS))),
     help="Comma-separated list of seeds to use",
 )
+parser.add_argument(
+    "--force",
+    choices=["train", "load", "none"],
+    default="train",
+    help="Force training or loading",
+)
 args = parser.parse_args()
 
 with tqdm(map(int, args.seeds.split(",")), desc="Seed") as pbar:
     for seed in pbar:
         cfg = ARGMAX_OF_5_CONFIG(seed, deterministic=False)
         pbar.set_postfix({"seed": seed, "cfg": cfg})
-        runtime, model = train_or_load_model(cfg)  # , force="train"
+        runtime, model = train_or_load_model(cfg, force=args.force)