(self, state)
| 52 | super(AdamW2, self).__init__(params, defaults) |
| 53 | |
| 54 | def __setstate__(self, state): |
| 55 | super(AdamW2, self).__setstate__(state) |
| 56 | for group in self.param_groups: |
| 57 | group.setdefault('amsgrad', False) |
| 58 | |
| 59 | @torch.no_grad() |
| 60 | def step(self, closure=None): |
nothing calls this directly
no outgoing calls
no test coverage detected