(self)
| 261 | return self._get_closed_form_lr() |
| 262 | |
| 263 | def _get_closed_form_lr(self): |
| 264 | warmup = 1 - self.warmup ** (self.last_epoch + 1) |
| 265 | return [warmup * base_lr for base_lr in self.base_lrs] |
| 266 | |
| 267 | |
| 268 | def stratified_uniform(shape, group=0, groups=1, dtype=None, device=None): |