Conv2DCustomBackpropInputOp 是否只支持 NHWC？答案

【问题标题】：Is it true that Conv2DCustomBackpropInputOp only supports NHWC?Conv2DCustomBackpropInputOp 是否只支持 NHWC？
【发布时间】：2019-11-17 02:09:22
【问题描述】：

问题是当我使用 TensorFlow 2.0.0 和 Python 3.7 来训练用 TensorFlow 1.0.0 和 Python 2.7 编写的 gan 时，我发现 Conv2DCustomBackpropInputOp 只支持 NHWC。是这样吗？

【问题讨论】：

data_format='channels_first' 好像坏了

标签： tensorflow computer-vision

【解决方案1】：

我遇到了这个错误，我只是通过将 CPU 更改为 GPU 来解决它！

因此，如果您在 CPU 上运行模型，请尝试将运行时更改为 CUDA，并使用您的 GPU。它可能会起作用。

【讨论】：

【解决方案2】：

您的输入数据的形状很可能是NCHW (Batch Size, Number of Channels, Height of the Image and Width of the Image).

将其形状更改为 NHWC (Batch Size, Height of the Image, Width of the Image and Number of Channels) 和 data_format='channels_last' 应该可以解决您的问题。

【讨论】：

我认为问题在于 Conv2D 是否仅支持 NHWC（而不支持 NCHW）。我遇到了同样的问题，很明显可以通过将订单更改为 NHWC 来解决，但是为什么我们需要这样做呢？ Conv2D 接受一个参数来指定输入顺序，所以我希望它在这两种情况下都能正常工作