igeng84
igeng84
> do not set the batch size to 1, you may try 2 or more > > a better choice would be apply "if prediction.shape == (1,5+1+len(classes)):" > instead of...
你可以留个联系方式，我这里有一些你说的资料，可以免费分享给你。
这不就是显卡内存不够了呗，人家用的是2080ti，你试试减小batch_size，或者减少数据试试。