run_hf_scaled_job.sh

#!/usr/bin/env bash
set -euo pipefail

python -m pip install --no-cache-dir \
  'numpy==1.26.4' \
  'pandas==2.2.3' \
  'scipy==1.14.1' \
  'scikit-learn==1.6.1' \
  'lightgbm==4.6.0' \
  'torchvision==0.21.0'

python /work/reproduce_ert_scaled.py \
  --output-dir /work/scaled_run \
  --data-dir /tmp/ert-data \
  --seeds 4 \
  --fashion-test 20000 \
  --base-epochs 3