ValueError: cannot copy sequence with size 84 to array axis with dimension 4 #892
Description
python -m baselines.run --alg=deepq --env=PongNoFrameskip-v4 --num_timesteps=1e5 --print_freq=10 --save_path=~/models/pong
python -m baselines.run --alg=deepq --env=PongNoFrameskip-v4 --num_timesteps=0 --load_path=~/models/pong --play
Loaded model from ~/models/pong
Running trained model
Traceback (most recent call last):
File "/Users/odats/anaconda3/lib/python3.7/runpy.py", line 193, in _run_module_as_main
"__main__", mod_spec)
File "/Users/odats/anaconda3/lib/python3.7/runpy.py", line 85, in _run_code
exec(code, run_globals)
File "/Users/odats/Documents/Projects/uku/uku-ml-project/baselines/baselines/run.py", line 244, in <module>
main(sys.argv)
File "/Users/odats/Documents/Projects/uku/uku-ml-project/baselines/baselines/run.py", line 228, in main
actions, _, _, _ = model.step(obs)
File "/Users/odats/Documents/Projects/uku/uku-ml-project/baselines/baselines/deepq/deepq.py", line 53, in step
return self._act([observation], **kwargs), None, None, None
File "/Users/odats/Documents/Projects/uku/uku-ml-project/baselines/baselines/deepq/build_graph.py", line 198, in act
return _act(ob, stochastic, update_eps)
File "/Users/odats/Documents/Projects/uku/uku-ml-project/baselines/baselines/common/tf_util.py", line 179, in <lambda>
return lambda *args, **kwargs: f(*args, **kwargs)[0]
File "/Users/odats/Documents/Projects/uku/uku-ml-project/baselines/baselines/common/tf_util.py", line 208, in __call__
self._feed_input(feed_dict, inpt, value)
File "/Users/odats/Documents/Projects/uku/uku-ml-project/baselines/baselines/common/tf_util.py", line 196, in _feed_input
feed_dict.update(inpt.make_feed_dict(value))
File "/Users/odats/Documents/Projects/uku/uku-ml-project/baselines/baselines/deepq/utils.py", line 38, in make_feed_dict
return {self._placeholder: adjust_shape(self._placeholder, data)}
File "/Users/odats/Documents/Projects/uku/uku-ml-project/baselines/baselines/common/tf_util.py", line 394, in adjust_shape
data = np.array(data)
ValueError: cannot copy sequence with size 84 to array axis with dimension 4