search

carpedm20 / pixel-rnn-tensorflow

in progress
MIT License
496 stars 129 forks source link

Image data format #13

Open yangj1e opened 7 years ago

yangj1e commented 7 years ago

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.