gpu_train.sh 301 字节
Newer Older
Y
yudongxu(许煜东) 已提交
1
CUDA_VISIBLE_DEVICES=0 python train.py --use_gpu 1\  
O
overlordmax 已提交
2 3 4 5
									   --epochs 100\  
									   --batch_size 64\  
									   --embed_size 12\  
									   --cpu_num 2\  
Y
yudongxu(许煜东) 已提交
6 7 8
									   --model_dir ./model_dir\
									   --train_data_path ./train_data\  
									   --vocab_path ./vocab/vocab_size.txt