Error Conv2DCustomBackpropFilterOp only supports NHWC

igul222 / improved_wgan_training

Code for reproducing experiments in "Improved Training of Wasserstein GANs"

MIT License

2.35k stars 668 forks source link

Error Conv2DCustomBackpropFilterOp only supports NHWC #87

Open ssinha89 opened 4 years ago

ssinha89 commented 4 years ago

When I run gan_mnist.py I get the error Conv2DCustomBackpropFilterOp only supports NHWC. Could you please mention which version of tensorflow is needed to run this code ? I have tried multiple versions, getting the same error.

KeluYao commented 4 years ago

tensorflow 1.8 works for me

qxm1995716 commented 4 years ago

You should install the tensorflow-GPU, check it.