(scheduler, num_steps=10)
| 37 | |
| 38 | |
| 39 | def unwrap_schedule(scheduler, num_steps=10): |
| 40 | lrs = [] |
| 41 | for _ in range(num_steps): |
| 42 | scheduler.step() |
| 43 | lrs.append(scheduler.get_lr()) |
| 44 | return lrs |
| 45 | |
| 46 | |
| 47 | def unwrap_and_save_reload_schedule(scheduler, num_steps=10): |
no test coverage detected