blob: cf5669e228cf6d413fdea335075c22225bb6debe (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
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
|
