#!/usr/bin/env bash
set -euo pipefail
cd "$(dirname "$0")"

export OMP_NUM_THREADS=2
PY=~/joulebeat-evolve/outreach/env/bin/python

mkdir -p data logs figs

if [ ! -f data/cifar10_test.parquet ]; then
  echo "Downloading CIFAR-10 test set (HF parquet mirror, torchvision's own host is too slow on this network)..."
  curl -o data/cifar10_test.parquet -L \
    "https://huggingface.co/datasets/uoft-cs/cifar10/resolve/main/plain_text/test-00000-of-00001.parquet"
fi

"$PY" attack_experiment.py \
  --grad-budgets 36,72 \
  --seeds 0,1,2 \
  --n-images 128 \
  --out results.json \
  --figs-dir figs \
  --data data/cifar10_test.parquet