Update run_iu_xray.sh

wjhou · Oct 10, 2023 · 084c309 · 084c309
1 parent 4397eb7
commit 084c309
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/script/run_iu_xray.sh b/script/run_iu_xray.sh
@@ -50,6 +50,7 @@ python3 -u ./src/run_ende.py \
     --trr $trr \
     --topk $topk \
     --outline_level $outline_level \
+    --beta $beta \
     --per_device_train_batch_size $per_device_train_batch_size \
     --per_device_eval_batch_size $per_device_eval_batch_size \
     --gradient_accumulation_steps $gradient_accumulation_steps \