I got error when I tried to run it with trained data and display-on

[wonchul-kim]

Hi!
First of all, I really thank you for your share!

I set all the setting that needed to run it.

I could implement the process to train a model for breakout
(python main.py --env_name=Breakout-v0 --is_train=True --display=True)

However, When i tried to run it as test and record
(python main.py --is_train=False --display=True)

I've got this error:

InvalidArgumentError (see above for traceback): CPU BiasOp only supports NHWC.
[[Node: prediction/l1/BiasAdd = BiasAdd[T=DT_FLOAT, data_format="NCHW", _device="/job:localhost/replica:0/task:0/cpu:0"](prediction/l1/Conv2D, prediction/l1/biases/read)]]

Could you please help me out?