tensorflow - Tensorflow 2 对象检测 API：无法将批量大小设置为 1

Question

如果我将其更改batchsize为1更快RCNN 640x640且ResNet 50作为主干，

我收到以下错误：

The 'global_batch_size' 1 is not divisible by 'num_replicas_in_sync' 4
How can I set the batchsize to 1?

标签： tensorflowtensorflow2.0tensorflow2.x

Answer 1

num_replicas_in_sync正如您可能想象的那样，您的问题是您实际上正在尝试使用 4 个视频卡/设备（

有两种解决方案：

1. 要么更改num_replicas_in_sync为 1。
2. 将 增加global_batch_size至可被 4 整除的值（4、8、12 等）