File size: 440 Bytes
45eaa9d | 1 2 3 4 5 6 7 8 9 | #!/usr/bin/env bash
set -euo pipefail
DATASET=${DATASET:-cifar10}; SEEDS=${SEEDS:-"7 17 27"}; TIERS=${TIERS:-"tiny small base large"}
mkdir -p results/scaling
for tier in $TIERS; do for seed in $SEEDS; do
python benchmarks/train_image.py --config configs/${tier}.json --dataset "$DATASET" --seed "$seed" --amp --output results/scaling/${DATASET}_${tier}_s${seed}.json
done; done
python scripts/summarize_scaling.py results/scaling/*.json
|