batch_size问题
有同学经常会提示batch_size不能整除,这时候你可以调整一下batch_size的大小,一般设置为1,2,4,8,16。如果都不行,那么你就要检查一下自己的代码和环境,当然也不排除你服务器只有三张显卡,我们服务器就是,你设置12看看,应该可以运行。
训练yolov5中遇到的问题
最新推荐文章于 2024-04-14 00:39:45 发布
batch_size问题
有同学经常会提示batch_size不能整除,这时候你可以调整一下batch_size的大小,一般设置为1,2,4,8,16。如果都不行,那么你就要检查一下自己的代码和环境,当然也不排除你服务器只有三张显卡,我们服务器就是,你设置12看看,应该可以运行。