-
Notifications
You must be signed in to change notification settings - Fork 0
/
train.sh
54 lines (51 loc) · 1.11 KB
/
train.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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
#!/bin/bash
# BART
python pipeline/train.py \
--data_dir ../data/dart/data/v1.1.1/ \
--dataset DART \
--linearize True \
--linearize_strategy 2 \
--run_id bart_v2 \
--model BART \
--batch_size 8 \
--use_mixed_precision True \
--num_epochs 5 \
--overwrite True \
--max_permutations 3
# T5
python pipeline/train.py \
--data_dir ../data/dart/data/v1.1.1/ \
--dataset DART \
--linearize True \
--linearize_strategy 2 \
--run_id t5_v2 \
--model T5 \
--batch_size 16 \
--use_mixed_precision True \
--num_epochs 5 \
--overwrite True \
--max_permutations 3
# # BERTShare
# python pipeline/train.py \
# --data_dir ../data/dart/data/v1.1.1/ \
# --dataset DART \
# --linearize True \
# --linearize_strategy 0 \
# --run_id bertshare_v0 \
# --model BERTShare \
# --batch_size 4 \
# --use_mixed_precision True \
# --num_epochs 1 \
# --overwrite True
# # RoBERTaShare
# python pipeline/train.py \
# --data_dir ../data/dart/data/v1.1.1/ \
# --dataset DART \
# --linearize True \
# --linearize_strategy 0 \
# --run_id robertashare_v0 \
# --model RoBERTaShare \
# --batch_size 16 \
# --use_mixed_precision True \
# --num_epochs 5 \
# --overwrite True