gaspersic/baselines
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Drop S and M args so that --play works (#636)

Browse Source
This commit is contained in:
Matthew Rahtz
2018-10-16 16:28:23 -07:00
committed by pzhokhov
parent a03dacd68d
commit ef96f3835b
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
baselines/deepq/deepq.py
View File

@@ -47,6 +47,9 @@ class ActWrapper(object):
return self._act(*args, **kwargs) return self._act(*args, **kwargs)
def step(self, observation, **kwargs): def step(self, observation, **kwargs):
# DQN doesn't use RNNs so we ignore states and masks
kwargs.pop('S', None)
kwargs.pop('M', None)
return self._act([observation], **kwargs), None, None, None return self._act([observation], **kwargs), None, None, None
def save_act(self, path=None): def save_act(self, path=None):