(config, optimizer, n_iter_per_epoch)
| 295 | |
| 296 | |
| 297 | def build_scheduler(config, optimizer, n_iter_per_epoch): |
| 298 | num_steps = int(config.epochs * n_iter_per_epoch) |
| 299 | warmup_steps = int(config.warmup_epochs * n_iter_per_epoch) |
| 300 | |
| 301 | lr_scheduler = CosineLRScheduler( |
| 302 | optimizer, |
| 303 | t_initial=num_steps, |
| 304 | lr_min=config.min_lr, |
| 305 | warmup_lr_init=config.warmup_lr, |
| 306 | warmup_t=warmup_steps, |
| 307 | cycle_limit=1, |
| 308 | t_in_epochs=False, |
| 309 | ) |
| 310 | return lr_scheduler |
| 311 | |
| 312 | def collate_fn(batch): |
| 313 | batch = list(zip(*batch)) |
nothing calls this directly
no outgoing calls
no test coverage detected