(self, params, lr=1e-3, betas=(0.9, 0.999), eps=1e-8, weight_decay=0)
| 119 | |
| 120 | class RAdam(Optimizer): |
| 121 | def __init__(self, params, lr=1e-3, betas=(0.9, 0.999), eps=1e-8, weight_decay=0): |
| 122 | defaults = dict(lr=lr, betas=betas, eps=eps, weight_decay=weight_decay) |
| 123 | super().__init__(params, defaults) |
| 124 | |
| 125 | def step(self, closure=None): |
| 126 | loss = None |