(self, optimizer, warmup_period, last_step=-1)
| 126 | """ |
| 127 | |
| 128 | def __init__(self, optimizer, warmup_period, last_step=-1): |
| 129 | _check_optimizer(optimizer) |
| 130 | group_count = len(optimizer.param_groups) |
| 131 | warmup_params = get_warmup_params(warmup_period, group_count) |
| 132 | super(ExponentialWarmup, self).__init__(optimizer, warmup_params, last_step) |
| 133 | |
| 134 | def warmup_factor(self, step, warmup_period): |
| 135 | return 1.0 - math.exp(-(step+1) / warmup_period) |
nothing calls this directly
no test coverage detected