echo "Starting training..."
uv sync --offline
source .venv/bin/activate
echo "Running model.py..."
python3 model.py \
    --data_from_hf="/gpfs1/scratch/Courses/ds/ds069/squad_v2_processed" \
    --model_name_or_path="/gpfs1/scratch/Courses/ds/ds069/bert-base-uncased" \
    --checkpoint_dir="checkpoints" \
    --ckpt_path="" \
    --max_epochs=2