diff options
Diffstat (limited to 'experiments/oral_a_bp_reference.sh')
| -rwxr-xr-x | experiments/oral_a_bp_reference.sh | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/experiments/oral_a_bp_reference.sh b/experiments/oral_a_bp_reference.sh new file mode 100755 index 0000000..cf5669e --- /dev/null +++ b/experiments/oral_a_bp_reference.sh @@ -0,0 +1,31 @@ +#!/usr/bin/env bash +set -euo pipefail + +variant="${1:-primary}" +device="${2:-cuda}" +mkdir -p results/oral_a_dev + +common=( + --mode bp --device "${device}" --depth 20 --width 16 --seed 0 --loader_seed 0 + --batch_size 128 --epochs 200 --val_examples 5000 --split_seed 2027 + --eval_split validation --eval_every 20 --augment_train 1 + --lr 0.1 --momentum 0.9 --normalization batchnorm +) + +case "${variant}" in + primary) + python3 experiments/conv_run.py "${common[@]}" \ + --lr_schedule step --lr_milestones 100,150 --lr_gamma 0.1 \ + --warmup_epochs 0 --weight_decay 1e-4 \ + --out results/oral_a_dev/bp_reference_primary.json + ;; + recovery) + python3 experiments/conv_run.py "${common[@]}" \ + --lr_schedule cosine --warmup_epochs 5 --weight_decay 5e-4 \ + --out results/oral_a_dev/bp_reference_recovery.json + ;; + *) + echo "variant must be primary or recovery" >&2 + exit 2 + ;; +esac |
