(self, optimizer, warmup_period, last_step=-1)
| 107 | """ |
| 108 | |
| 109 | def __init__(self, optimizer, warmup_period, last_step=-1): |
| 110 | _check_optimizer(optimizer) |
| 111 | group_count = len(optimizer.param_groups) |
| 112 | warmup_params = get_warmup_params(warmup_period, group_count) |
| 113 | super(LinearWarmup, self).__init__(optimizer, warmup_params, last_step) |
| 114 | |
| 115 | def warmup_factor(self, step, warmup_period): |
| 116 | return min(1.0, (step+1) / warmup_period) |
nothing calls this directly
no test coverage detected