Super-mario-bros-A3C-pytorch
Super-mario-bros-A3C-pytorch copied to clipboard
implementation of optimizer
Sorry, I refered to google but still confused about the GlabalAdam optimizer. Can someone give me a HELP??
for group in self.param_groups:
for p in group['params']:
state = self.state[p]
state['step'] = 0
state['exp_avg'] = torch.zeros_like(p.data)
state['exp_avg_sq'] = torch.zeros_like(p.data)
state['exp_avg'].share_memory_()
state['exp_avg_sq'].share_memory_()