(self, optimizer)
| 73 | return outputs |
| 74 | |
| 75 | def build_scheduler(self, optimizer): |
| 76 | return torch.optim.lr_scheduler.StepLR(optimizer, hparams['decay_steps'], gamma=0.5) |
| 77 | |
| 78 | def optimizer_step(self, epoch, batch_idx, optimizer, optimizer_idx): |
| 79 | if optimizer is None: |
nothing calls this directly
no outgoing calls
no test coverage detected