diff --git a/justfile b/justfile index 85825c8..85c5699 100644 --- a/justfile +++ b/justfile @@ -4,13 +4,6 @@ default: #!/bin/bash set -x - uv run python nbs/train.py gemma1b-24gb - uv run python nbs/train.py gemma1b-24gb --no_rot_v - uv run python nbs/train.py gemma1b-24gb --lr=1e-5 - - cd ../AntiPaSTO2-asym-coh - just - uv run pytest tests/test_train.py::test_train_rnd -v uv run pytest tests/test_train.py::test_train_tiny -v