--- 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 + +