cfahlgren1 HF Staff commited on
Commit
0de7556
·
verified ·
1 Parent(s): c97f9cc

fetch the shared eval from v2/ even when uploading to an experiment folder

Browse files
Files changed (1) hide show
  1. v2/train.py +3 -2
v2/train.py CHANGED
@@ -577,7 +577,8 @@ def parse_args():
577
  parser.add_argument("--eval-pauses", type=int, default=1000)
578
  parser.add_argument("--log-every", type=int, default=50)
579
  parser.add_argument("--repo", default="cfahlgren1/tinynext")
580
- parser.add_argument("--folder", default="v2")
 
581
  parser.add_argument("--name", default="tinynext-v2")
582
  parser.add_argument("--trackio-space", default="cfahlgren1/tinynext-trackio")
583
  parser.add_argument("--stage", choices=("all", "prepare", "train"), default="all",
@@ -620,7 +621,7 @@ def main():
620
  torch.backends.cudnn.allow_tf32 = True
621
  output = HERE / "output"
622
  output.mkdir(exist_ok=True)
623
- ev = load_eval_module(args.repo, args.folder)
624
  weights = {k: float(v) for k, v in (part.split("=") for part in args.mix.split(","))}
625
  print(f"device {device}, {cpus()} CPUs, mix {weights}", flush=True)
626
  tracker = start_tracker(args)
 
577
  parser.add_argument("--eval-pauses", type=int, default=1000)
578
  parser.add_argument("--log-every", type=int, default=50)
579
  parser.add_argument("--repo", default="cfahlgren1/tinynext")
580
+ parser.add_argument("--folder", default="v2", help="Where this run's outputs are uploaded")
581
+ parser.add_argument("--eval-folder", default="v2", help="Where the shared eval.py and eval_set.json live")
582
  parser.add_argument("--name", default="tinynext-v2")
583
  parser.add_argument("--trackio-space", default="cfahlgren1/tinynext-trackio")
584
  parser.add_argument("--stage", choices=("all", "prepare", "train"), default="all",
 
621
  torch.backends.cudnn.allow_tf32 = True
622
  output = HERE / "output"
623
  output.mkdir(exist_ok=True)
624
+ ev = load_eval_module(args.repo, args.eval_folder)
625
  weights = {k: float(v) for k, v in (part.split("=") for part in args.mix.split(","))}
626
  print(f"device {device}, {cpus()} CPUs, mix {weights}", flush=True)
627
  tracker = start_tracker(args)