search

miemie2013 / Keras-YOLOv4

yolov4 42.0% mAP.ppyolo 45.1% mAP.
446 stars 127 forks source link

单机多卡应该怎么操作 #11

Open hx358031364 opened 4 years ago

hx358031364 commented 4 years ago

model = keras.models.Model([model_body.input, *y_true], loss_list) parallel_model = multi_gpu_model(model, 2) 添加了multi_gpu_model，将后面所有model换成parallel_model，在保存模型的时候用model.save，但是不起作用，还是只是一张卡

oym050922021 commented 4 years ago

大神，你好，我也很疑惑如何实现单机多卡运行代码，非常感谢！

oym050922021 commented 4 years ago

model = keras.models.Model([model_body.input, *y_true], loss_list) parallel_model = multi_gpu_model(model, 2) 添加了multi_gpu_model，将后面所有model换成parallel_model，在保存模型的时候用model.save，但是不起作用，还是只是一张卡

hi,你实现单机多卡运行了吗