(opti, config)
| 25 | |
| 26 | |
| 27 | def build_lambda_sche(opti, config): |
| 28 | if config.get('decay_step') is not None: |
| 29 | lr_lbmd = lambda e: max(config.lr_decay ** (e / config.decay_step), config.lowest_decay) |
| 30 | scheduler = torch.optim.lr_scheduler.LambdaLR(opti, lr_lbmd) |
| 31 | else: |
| 32 | raise NotImplementedError() |
| 33 | return scheduler |
| 34 | |
| 35 | |
| 36 | def build_lambda_bnsche(model, config): |