SAMAD-fork/train.sh at main · howard9199/SAMAD-fork · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
#!/bin/bash

# ======================= Intermediate =======================
form_id='1964' # 1572, 1764, 1766, 1964(Main)
date='0110'
LEARNING_RATE='1e-5' # 1e-3
module_type="SAMAD_QWKloss" # content_only, content_langUse, SAMAD, SAMAD_wav2vec

train_file="/share/nas165/peng/thesis_project/(ok)SAMAD_06/data/LTTC_Intermediate/Unseen_${form_id}/train_${form_id}_0520.csv"
dev_file="/share/nas165/peng/thesis_project/(ok)SAMAD_06/data/LTTC_Intermediate/Unseen_${form_id}/dev_${form_id}_0520.csv"
#train_file="/datas/store163/howard/samad/SAMAD/picture-description/cleaned_train_1764_0520_merged.csv"
#dev_file="/datas/store163/howard/samad/SAMAD/picture-description/cleaned_dev_1764_0520_merged.csv"

exp_dir="./exp/LTTC-Intermediate/IS-${form_id}/${module_type}_${date}_${LEARNING_RATE}_roundown"


# train: train.py, train_softlabel, train_wav2vec, train_subModel, pretrained Model
#
#CUDA_VISIBLE_DEVICES=0 python3 ./models/train_subModel.py   --train_file "$train_file" \
#                        --dev_file  "$dev_file" \
#                        --output_dir "$exp_dir"  \
#                        --cuda_id 0 \
#                        --train_epochs 8  \
#                        --train_batch  2  \
#                        --eval_batch   2 \
#                        --grad_acc     4 \
#                        --learning_rate $LEARNING_RATE \
CUDA_VISIBLE_DEVICES=0 python3 ./models/train_subModel.py  \
                        --output_dir "$exp_dir"  \
                        --cuda_id 0 \
                        --train_epochs 32  \
                        --train_batch  64  \
                        --eval_batch   64 \
                        --grad_acc     1 \
                        --learning_rate $LEARNING_RATE \

# ======================= High-Intermediate =======================
# form_id = '1962' # 1731, 1801, 1862, 1962(Main)
# train_file='/share/nas165/peng/thesis_project/SAMAD_06/data/LTTC_HI/Unseen_${form_id}/train_${form_id}_0520.csv'
# dev_file='/share/nas165/peng/thesis_project/SAMAD_06/data/LTTC_HI/Unseen_${form_id}/dev_${form_id}_0520.csv'

# exp_dir="./exp/LTTC-Intermediate/IS-${form_id}/${module_type}_${date}"
# LEARNING_RATE='1e-3' # 1e-3

# CUDA_VISIBLE_DEVICES=1 python3 ./models/train.py   --train_file "$train_file" \
#                         --dev_file  "$dev_file" \
#                         --output_dir "$exp_dir"  \
#                         --cuda_id 1 \
#                         --train_epochs 8  \
#                         --train_batch  8  \
#                         --eval_batch   8 \
#                         --grad_acc     4 \
#                         --learning_rate $LEARNING_RATE \