DouZero
/
Douzero_Resnet
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
Douzero_Resnet/cuda_train.cmd

1 line
95 B
Batchfile
Raw Permalink Normal View History

修复CPU与GPU不能同时训练的BUG，退出时自动存档
2021-12-19 20:00:42 +08:00
python train.py --load_model --batch_size 4 --learning_rate 0.0003 --enable_onnx --num_actors 1