train.sh 654 字节
Newer Older
1
export CUDA_VISIBLE_DEVICES=4,5,6,7
Y
Yibing Liu 已提交
2
python -u  ../../train.py --train_feature_lst data/train_feature.lst \
3 4 5
                   --train_label_lst data/train_label.lst \
                   --val_feature_lst data/val_feature.lst \
                   --val_label_lst data/val_label.lst \
6
                   --mean_var data/global_mean_var \
7
                   --checkpoints checkpoints \
8 9
                   --frame_dim 80  \
                   --class_num 3040  \
Y
Yibing Liu 已提交
10
                   --print_per_batches 100 \
11
                   --infer_models '' \
Y
Yibing Liu 已提交
12
                   --batch_size 16 \
13
                   --learning_rate 6.4e-5 \
14
                   --parallel