0

错误可能源于输入操作。连接到节点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

4

2 回答 2

2

当我遇到这个问题时，我正在使用 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 版本。

我希望我的回答会有所帮助。

于 2021-11-08T18:20:50.727 回答
0

当我在 colab 上使用efficientDet_D0 训练对象检测模型时遇到了同样的问题，解决我的问题的方法只是更改为 TPU 运行时。同样正如@Ahmad Abubakar 提到的那样，您应该尝试将批量减小到更小的值。我发现任何小于 10 的东西最适合我

于 2021-11-20T21:31:30.237 回答