函数调用堆栈_dummy_computation_fn，我在训练我的对象检测模型时得到这个错误 &g

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 -升级tensorflow-gpu==2.4.1，它工作了。另一个建议是卸载TensorFlow和tensoflow-gpu，然后重启运行时，安装你想要的TensorFlow版本。

可能的解决方案:

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

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

Answer 2

当我在colab上使用efficientDet_D0训练对象检测模型时，我遇到了同样的问题，解决我的问题的方法是简单地更改为TPU运行时。也正如@Ahmad Abubakar提到的，你应该试着减少你的批量大小。我发现任何少于10的东西对我来说都是最好的