fetch the shared eval from v2/ even when uploading to an experiment folder
Browse files- 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.
|
| 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)
|