-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathfinetune_seq2seq.sh
35 lines (32 loc) · 1.1 KB
/
finetune_seq2seq.sh
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
GPU=1 \
MODEL_TYPE="seq2seq" \
DATASET="sri/py150" \
DECODER_ONLY="true" \
EPOCHS=10 \
CHECKPOINT="pretrain_csn_hidden_identity_512_512_random/ckpt_pretrain_ep0100_step0075000.pth" \
MODEL_NAME="finetuned_csn_h-sz-512-identity-random-fz" \
time make finetune-contracode
# GPU=1 \
# MODEL_TYPE="seq2seq" \
# DATASET="sri/py150" \
# DECODER_ONLY="true" \
# EPOCHS=10 \
# CHECKPOINT="pretrain_csn_hidden_identity_512_512_adv/ckpt_pretrain_ep0100_step0075000.pth" \
# MODEL_NAME="finetuned_csn_h-sz-512-identity-adv-fz" \
# time make finetune-contracode
# GPU=1 \
# MODEL_TYPE="seq2seq" \
# DATASET="sri/py150" \
# DECODER_ONLY="false" \
# EPOCHS=10 \
# CHECKPOINT="pretrain_csn_hidden_identity_512_512_random/ckpt_pretrain_ep0100_step0075000.pth" \
# MODEL_NAME="finetuned_csn_h-sz-512-identity-random" \
# time make finetune-contracode
# GPU=1 \
# MODEL_TYPE="seq2seq" \
# DATASET="sri/py150" \
# DECODER_ONLY="false" \
# EPOCHS=10 \
# CHECKPOINT="pretrain_csn_hidden_identity_512_512_adv/ckpt_pretrain_ep0100_step0075000.pth" \
# MODEL_NAME="finetuned_csn_h-sz-512-identity-adv" \
# time make finetune-contracode