(self)
| 52 | return self.common_step(batch, batch_idx) |
| 53 | |
| 54 | def configure_optimizers(self): |
| 55 | optimizer = optim.SGD(self.model.parameters(), self.lr, |
| 56 | momentum=self.momentum, |
| 57 | weight_decay=self.weight_decay) |
| 58 | |
| 59 | scheduler = optim.lr_scheduler.MultiStepLR(optimizer, milestones=[30, 60, 90], gamma=0.1) |
| 60 | return [optimizer], [scheduler] |
| 61 |
nothing calls this directly
no outgoing calls
no test coverage detected