Diff of /scripts/train_decoding.sh [000000] .. [66af30]

Switch to side-by-side view

--- a
+++ b/scripts/train_decoding.sh
@@ -0,0 +1,14 @@
+python3 train_decoding.py --model_name BrainTranslator \
+    --task_name task1_task2_taskNRv2 \
+    --one_step \
+    --pretrained \
+    --not_load_step1_checkpoint \
+    --num_epoch_step1 20 \
+    --num_epoch_step2 30 \
+    -lr1 0.00005 \
+    -lr2 0.0000005 \
+    -b 32 \
+    -s ./checkpoints/decoding \
+    -cuda cuda:0
+
+