pixel-rnn-tensorflow icon indicating copy to clipboard operation
pixel-rnn-tensorflow copied to clipboard

Image data format

Open yangj1e opened this issue 7 years ago • 0 comments

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.

yangj1e avatar Mar 10 '17 11:03 yangj1e