#!/usr/bin/env bash
cd "$(dirname "$0")"
export PYTHONPATH="$PWD" OMP_NUM_THREADS=8
exec ./.venv/bin/python -u train/train_sft2.py \
  --base ckpt/nlp --data data/sft_mix_v2.jsonl --tok data/tokenizer.json \
  --ckpt ckpt/v2 --epochs 3 --batch 8 --seq 256 --lr 2e-5 \
  --eval-every 25 --log-every 25 --threads 8