forked from researchmm/TTSR
-
Notifications
You must be signed in to change notification settings - Fork 0
/
train.sh
executable file
·46 lines (44 loc) · 1.49 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
### training TTSR
python main.py --save_dir ./train/CUFED/TTSR \
--reset True \
--log_file_name train.log \
--num_gpu 1 \
--num_workers 9 \
--dataset CUFED \
--dataset_dir /home/v-fuyang/Data/CUFED/ \
--n_feats 64 \
--lr_rate 1e-4 \
--lr_rate_dis 1e-4 \
--lr_rate_lte 1e-5 \
--rec_w 1 \
--per_w 1e-2 \
--tpl_w 1e-2 \
--adv_w 1e-3 \
--batch_size 9 \
--num_init_epochs 2 \
--num_epochs 50 \
--print_every 600 \
--save_every 10 \
--val_every 10
# ### training TTSR-rec
# python main.py --save_dir ./train/CUFED/TTSR-rec \
# --reset True \
# --log_file_name train.log \
# --num_gpu 1 \
# --num_workers 9 \
# --dataset CUFED \
# --dataset_dir /home/v-fuyang/Data/CUFED/ \
# --n_feats 64 \
# --lr_rate 1e-4 \
# --lr_rate_dis 1e-4 \
# --lr_rate_lte 1e-5 \
# --rec_w 1 \
# --per_w 0 \
# --tpl_w 0 \
# --adv_w 0 \
# --batch_size 9 \
# --num_init_epochs 0 \
# --num_epochs 200 \
# --print_every 600 \
# --save_every 10 \
# --val_every 10