问题描述
我正在使用tensorflow 2.2训练MIL模型。该模型在每个袋（批次）中具有不同数量的图像，范围从每批次18到28张图像（512,512,3）。我正在使用2个GPU 2080 TI。我在这里看到的问题是遇到小批量时。如果批处理量较小，则所有数据或批处理数据仅被馈送到一台GPU机器中，而第二个GPU则不接收任何批处理数据，从而产生了错误。 有什么解决方案可以在培训和相应地使用gpu时处理各种批处理集？
解决方法
暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！
如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@）