forked from santi-pdp/pase
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtrain_dropout.sh
27 lines (25 loc) · 861 Bytes
/
train_dropout.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
#!/bin/bash
#SBATCH -J dropout
#SBATCH -o log/dropout_32000
#SBATCH --mem=32GB
#SBATCH -t 5-00:00:00
#SBATCH -n 1
#SBATCH -c 8
#SBATCH -p gpu
#SBATCH --gres=gpu -C K80
module load cuda
module load anaconda
source activate humor
conda info -e
nvidia-smi
python -u train.py --batch_size 32 --epoch 150 --num_workers 8 \
--save_path /scratch/jzhong9/model_ckpt/dropout_32000 \
--net_cfg cfg/workers.cfg \
--fe_cfg cfg/PASE_dense.cfg \
--do_eval --data_cfg /scratch/jzhong9/data/LibriSpeech_50h/librispeech_data_50h.cfg --min_lr 0.0005 --fe_lr 0.001 \
--data_root /scratch/jzhong9/data/LibriSpeech_50h/wav_sel \
--stats /scratch/jzhong9/data/LibriSpeech_50h/librispeech_50h_stats.pkl \
--log_freq 100 \
--backprop_mode dropout --dropout_rate 0.5 \
--tensorboard False \
--chunk_size 32000