search

devsisters / DQN-tensorflow

Tensorflow implementation of Human-Level Control through Deep Reinforcement Learning
MIT License
2.48k stars 764 forks source link

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

Closed wonchul-kim closed 7 years ago

wonchul-kim commented 7 years ago

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?