pixel-rnn-tensorflow
pixel-rnn-tensorflow copied to clipboard
Image data format
In network.py
if conf.use_gpu:
data_format = "NHWC"
else:
data_format = "NCHW"
I think NCHW
is preferred for GPU, while for CPU, we should use NHWC
.