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

自己的数据集训练YOLOV4模型

第一个epoch出现以下报错

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