Open xtendai opened 5 years ago
We are getting the following error :
InvalidArgumentError (see above for traceback): Conv2DCustomBackpropInputOp only supports NHWC. [[{{node gradients/Discriminator.4_2/Conv2D_grad/Conv2DBackpropInput}} = Conv2DBackpropInput[T=DT_FLOAT, _class=["loc:@gradi...pendency_1"], data_format="NCHW", dilations=[1, 1, 1, 1], padding="SAME", strides=[1, 1, 2, 2], use_cudnn_on_gpu=true, _device="/job:localhost/replica:0/task:0/device:CPU:0"](ConstantFolding/gradients_2/Discriminator.4_2/Conv2D_grad/ShapeN-matshapes-0, Discriminator.4/Discriminator.4.Filters/read, gradients/AddN_2)]]
How do we change the format to NHWC to make it work?
We are using the image dump from the Google Drive link above - and all images have been resized to 64x64.
Versions: Python 3.6.6 ; Tensorflow 1.11 and Nvidia GPU 1080
Thanks
use python 3.5.6 in anaconda
lanyuue
commented
5 years ago lanyuue
commented
5 years ago
We are getting the following error :
InvalidArgumentError (see above for traceback): Conv2DCustomBackpropInputOp only supports NHWC. [[{{node gradients/Discriminator.4_2/Conv2D_grad/Conv2DBackpropInput}} = Conv2DBackpropInput[T=DT_FLOAT, _class=["loc:@gradi...pendency_1"], data_format="NCHW", dilations=[1, 1, 1, 1], padding="SAME", strides=[1, 1, 2, 2], use_cudnn_on_gpu=true, _device="/job:localhost/replica:0/task:0/device:CPU:0"](ConstantFolding/gradients_2/Discriminator.4_2/Conv2D_grad/ShapeN-matshapes-0, Discriminator.4/Discriminator.4.Filters/read, gradients/AddN_2)]]
How do we change the format to NHWC to make it work?
We are using the image dump from the Google Drive link above - and all images have been resized to 64x64.
Versions: Python 3.6.6 ; Tensorflow 1.11 and Nvidia GPU 1080
Thanks