我正在玩Karras等人的Progressive Growing of Gans网络。 （NVIDIA）。我使用NCHW模式对所有卷积as seen here使用不同的数据集在2x 1080 Ti卡上训练了网络。
现在我有一个训练有素的模型，我想使用项目自述文件中的代码段到import the trained network-在导入和使用预训练的网络部分中提到。 / p>
但是，当我尝试仅在具有CPU的PC上运行它时，网络将失败并显示以下错误：
InvalidArgumentError (see above for traceback): Conv2DCustomBackpropInputOp only supports NHWC.
[[node Gs/Run/Gs/cond/8x8/Conv0_up/conv2d_transpose (defined at <string>:93) ]]
我知道CPU没有实现NCHW，但是我想知道如何解决这个问题？我看到一些我不想要的选项：
首先想到的是，即使网络在NCHW中进行了训练，也仅使用NHWC模式生成新图像。但是，如果我是正确的话，这将破坏该图层的权重，因为它是在NCHW上训练的。
我不想以NHWC模式训练整个网络，因为它应该比NCHW慢一些？我没有使用NVidia的cuDNN，这是否意味着这两种模式与according to this的速度相同？
我还能做什么？谢谢！