summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/eval.py1
1 files changed, 0 insertions, 1 deletions
diff --git a/src/eval.py b/src/eval.py
index 7dcd289..50ec40e 100644
--- a/src/eval.py
+++ b/src/eval.py
@@ -147,7 +147,6 @@ def main() -> None:
ap = argparse.ArgumentParser(description="prime-grokking eval")
ap.add_argument("model")
ap.add_argument("seed")
- ap.add_argument("--ckpt", default="last.pt")
ap.add_argument("--runs-dir", default="runs")
a = ap.parse_args()
out_dir = os.path.join(a.runs_dir, a.model, f"seed{a.seed}")