我是大勤快 2022-06-14 10:57 采纳率: 0%
浏览 22

关于#人工智能#的问题:自己的数据集训练YOLOV4模型

自己的数据集训练YOLOV4模型
第一个epoch出现以下报错

img

  • 写回答

1条回答 默认 最新

  • 爱晚乏客游 2022-06-14 15:58
    关注

    img


    显存不够,缩小输入或者batch size.另外上面那个报错看下是不是显存的问题,要么就是tf1.x和2.x版本的问题,对不上就各种问题。yolov4建议走darknet,不想走这个的话建议用pytorch,早换早轻松

    评论

报告相同问题?

问题事件

  • 创建了问题 6月14日