Super-mario-bros-A3C-pytorch icon indicating copy to clipboard operation
Super-mario-bros-A3C-pytorch copied to clipboard

implementation of optimizer

Open QingyvHan opened this issue 3 years ago • 0 comments

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_()

QingyvHan avatar Dec 10 '20 11:12 QingyvHan