diff options
| -rwxr-xr-x | experiments/rain_ep_upfront_calibration_r5.sh | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/experiments/rain_ep_upfront_calibration_r5.sh b/experiments/rain_ep_upfront_calibration_r5.sh index d759111..7b997df 100755 --- a/experiments/rain_ep_upfront_calibration_r5.sh +++ b/experiments/rain_ep_upfront_calibration_r5.sh @@ -5,7 +5,7 @@ ROOT=/home/yurenh2/sdil AUTHOR=/scratch/yurenh2/energy-based-learning PYTHON=/scratch/yurenh2/venvs/burstccn/bin/python PROFILE="$ROOT/results/physical_bias/p0_state_dependence.json" -OUT="$ROOT/results/ep_bias/centered_r5_upfront" +OUT="${RAIN_EP_OUT:-$ROOT/results/ep_bias/centered_r5_upfront}" mkdir -p "$OUT" cd "$AUTHOR" @@ -14,6 +14,7 @@ run_cell() { local probes=$2 local mode=$3 local tag=$4 + local predictor_kind=$5 CUBLAS_WORKSPACE_CONFIG=:4096:8 CUDA_VISIBLE_DEVICES="$gpu" "$PYTHON" \ "$ROOT/experiments/rain_ep_bias_train.py" \ --author-root "$AUTHOR" --device cuda \ @@ -21,6 +22,7 @@ run_cell() { --beta-policy centered --beta-value 0.25 \ --mode "$mode" --bias-ratio 1 \ --dillavou-profile-json "$PROFILE" --predictor-rate 1 \ + --predictor-kind "$predictor_kind" \ --dillavou-calibration-steps "$probes" --neutral-cadence 0 \ --epochs 1 --schedule-epochs 100 \ --train-limit 10000 --test-limit 2000 --batch-size 128 \ @@ -33,9 +35,10 @@ run_cell() { probe_budgets=(2 4 8 16) gpu=0 for probes in "${probe_budgets[@]}"; do - run_cell "$gpu" "$probes" constant "probes_${probes}_intercept" & + run_cell "$gpu" "$probes" constant "probes_${probes}_intercept" nlms & gpu=$((gpu + 1)) - run_cell "$gpu" "$probes" innovation "probes_${probes}_sdil" & + run_cell "$gpu" "$probes" innovation "probes_${probes}_sdil" \ + "${SDIL_PREDICTOR_KIND:-nlms}" & gpu=$((gpu + 1)) done wait |