(config, checkpoint_manager)
| 101 | |
| 102 | |
| 103 | def create_rampup_manager(config, checkpoint_manager): |
| 104 | if not config.enable_rampup_batch_size: |
| 105 | return None |
| 106 | |
| 107 | # Current step default as -1 if no checkpoint exists |
| 108 | current_step = -1 |
| 109 | if checkpoint_manager and checkpoint_manager.latest_step(): |
| 110 | current_step = checkpoint_manager.latest_step() |
| 111 | |
| 112 | return RampupBatchManager(config, current_step) |
no test coverage detected