train_cpu.sh 325 字节
Newer Older
O
overlordmax 已提交
1 2 3 4 5 6 7 8 9
python train.py --epochs 40 \
                --batch_size 40 \
                --use_gpu 0 \
                --train_data_path 'train_data/train_data.csv' \
                --model_dir 'model_dir' \
                --hidden1_units 75 \
                --hidden2_units 50 \
                --hidden3_units 25