mobilenetv3-yolov3 icon indicating copy to clipboard operation
mobilenetv3-yolov3 copied to clipboard

Published 20 hours ago • verified publisher icon tanluren

train gpu4

Open MichaelCong opened this issue 5 years ago • 2 comments

python train.py --gpu_num GPU_NUM 4 why only one-gpu have worked?thanks

MichaelCong avatar Aug 17 '19 16:08 MichaelCong

python train.py --gpu_num 4 为什么无法使用GPU进行训练？是需要怎么修改一下吗？非常感谢你，谢谢

MichaelCong avatar Aug 18 '19 07:08 MichaelCong

Hi ,you can wrap the model with keras.utils.multi_gpu_model in train.py ,refer to the example in yolo.py

tanluren avatar Aug 18 '19 14:08 tanluren