函数调用堆栈_dummy_computation_fn，我在训练对象检测模型时遇到此错误

Question

错误可能源自输入操作。 连接到节点 ResNet101V1_FPN/model/conv1_conv/Conv2D 的输入源操作： ResNet101V1_FPN/model/lambda/Pad（定义于/local/lib/python3.7/dist-packages/object_detection/models/keras_models/resnet_v1.py:51）

函数调用栈： _dummy_computation_fn

Answer 1

遇到这个问题时，我正在使用 Tensorflow-gpu==2.6.0。经过搜索并看到很多建议。 我相信 TF 2.6.0 没有与 CuDNN 8 兼容的版本，因此我将 TF 降级到 2.4.1

!pip --upgrade tensorflow-gpu==2.4.1

并且它有效。 另一个建议是卸载 TensorFlow 和 tensoflow-gpu，然后重新启动运行时，安装您想要的 TensorFlow 版本。

可能的解决方案：

1. 将 Batch_size 从 64 减少到 8 或 4 或更低。
2. 重新启动运行时并卸载所有 Tensorflow-gpu 版本。

希望我的回答对你有帮助。

Answer 2

当我在 colab 上使用 effectiveDet_D0 训练对象检测模型时遇到了同样的问题，解决我的问题的方法很简单，就是更改为 TPU 运行时。另外，正如 @Ahmad Abubakar 提到的，你应该尝试将批量大小减小到更小。我发现小于 10 的任何内容都最适合我