如果图片过大,需要适当缩小batch_size的值,否则使用GPU时可能超出其缓存大小而报错

相关文章: