ptan
ptan copied to clipboard
Examples under ptan/samples are outdated
It seems that the examples under ptan/samples are outdated. For instance, the code for creating agent in dqn_expreplay.py does not match the current definition
agent = ptan.agent.DQNAgent(model, action_selector), cuda=cuda_enabled)
While in the current class definition, the creator arguments are mismatched:
class DQNAgent(BaseAgent):
def __init__(self, dqn_model, action_selector, device="cpu", preprocessor=default_states_preprocessor):
self.dqn_model = dqn_model
self.action_selector = action_selector
self.preprocessor = preprocessor
self.device = device
Can we update the samples as well? Thanks.