https://blog.csdn.net/hzh_0000/article/details/80393759
参考上面的博客,虽然没有解决问题,但是给了我启发。
【解决方法】
样本的数量要求能够整除batchsize,即num%batch_size==0
例如我的FLAGS.batch_size = 6 gpu_num =2 batch_size = 2*6 =12 要求样本数量num必须是12的倍数,样本数多一个或者少一个都不行
https://blog.csdn.net/hzh_0000/article/details/80393759
参考上面的博客,虽然没有解决问题,但是给了我启发。
【解决方法】
样本的数量要求能够整除batchsize,即num%batch_size==0
例如我的FLAGS.batch_size = 6 gpu_num =2 batch_size = 2*6 =12 要求样本数量num必须是12的倍数,样本数多一个或者少一个都不行