scripts/train_baselines.sh

#!/usr/bin/env bash
set -euo pipefail
cd /home/ningyuxi/workspace/latent-rubric
export PYTHONPATH=. CUDA_DEVICE_ORDER=PCI_BUS_ID CUDA_VISIBLE_DEVICES=GPU-2a9c799b-ab9e-c677-db0e-379f480cb00f OMP_NUM_THREADS=4 OPENBLAS_NUM_THREADS=2
for method in scalar latent ortho attribute; do
  .venv/bin/python -u scripts/train_heads.py --method "$method" "$@"
done
.venv/bin/python -u scripts/evaluate_heads.py