Keras-YOLOv4 icon indicating copy to clipboard operation
Keras-YOLOv4 copied to clipboard

多GPU训练遇到的问题

Open xulu666666 opened this issue 4 years ago • 4 comments

请问这个训练可以使用多GPU训练吗?我单独GPU可以,但是多个就不行了,错误如下。不知道是我这边代码的问题还是yolo4还没有实现多GPU训练 image

xulu666666 avatar Nov 24 '20 02:11 xulu666666

我也遇到这个问题了,不知道楼主解决了吗

ppy2017 avatar Nov 26 '20 14:11 ppy2017

我现在正在训练,如果训练完成后模型确实可以用,我再更新到issue里。

------------------ 原始邮件 ------------------ 发件人: "miemie2013/Keras-YOLOv4" <[email protected]>; 发送时间: 2020年11月26日(星期四) 晚上10:10 收件人: "miemie2013/Keras-YOLOv4"<[email protected]>; 抄送: "hit_leisurewewe.com"<[email protected]>;"Author"<[email protected]>; 主题: Re: [miemie2013/Keras-YOLOv4] 多GPU训练遇到的问题 (#41)

我也遇到这个问题了,不知道楼主解决了吗

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.

xulu666666 avatar Nov 27 '20 02:11 xulu666666

@ppy2017 我这个不行,训练完了模型只能多GPU预测,保存不了一个模型

xulu666666 avatar Nov 27 '20 07:11 xulu666666

@ppy2017 我这个不行,训练完了模型只能多GPU预测,保存不了一个模型

我就是怎么按百度方法改都不行。

ppy2017 avatar Nov 27 '20 07:11 ppy2017