(optimizer: Optimizer, num_warmup_steps: int, num_training_steps: int, final_ratio: float)
| 109 | |
| 110 | |
| 111 | def get_iter_exponential_schedule(optimizer: Optimizer, num_warmup_steps: int, num_training_steps: int, final_ratio: float): |
| 112 | lr_func = IterExponential(total_iter_length=num_training_steps, final_ratio=final_ratio, warmup_steps=num_warmup_steps) |
| 113 | return LambdaLR(optimizer=optimizer, lr_lambda=lr_func) |
| 114 | |
| 115 | |
| 116 | if __name__ == "__main__": |
no test coverage detected