mrzixi / sentiprompt Goto Github PK

task="penga/14lap penga/14res penga/15res penga/16res pengb/14lap pengb/14res pengb/15res pengb/16res lcx/14lap lcx/14res lcx/15res lcx/16res"
s="544 3210 8 5678 744"
for SEED in $s
do
for dataset in $task
do
LOG_DIR=/mnt/f/yj/senti/logs_prompt_type3bs16/$dataset/$SEED
python train.py --dataset_name $dataset \
                                       --log_dir $LOG_DIR \
                                       --batch_size 32 \
                                       --lr 5e-5 \
                                       --n_epochs 50 \
                                       --save_model $LOG_DIR/models \
                                       --prompt type3 \
                                       --seed $SEED
done
done