summaryrefslogtreecommitdiff
path: root/experiments/oral_a_bp_reference.sh
diff options
context:
space:
mode:
Diffstat (limited to 'experiments/oral_a_bp_reference.sh')
-rwxr-xr-xexperiments/oral_a_bp_reference.sh31
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