(self)
| 306 | return metrics |
| 307 | |
| 308 | def configure_optimizers(self): |
| 309 | optimizer, self.trainable_param_names = get_optimizer(self.model, self.config) |
| 310 | scheduler = get_scheduler(optimizer, self.config) |
| 311 | return { |
| 312 | "optimizer": optimizer, |
| 313 | "lr_scheduler": { |
| 314 | "scheduler": scheduler, |
| 315 | "interval": "step", |
| 316 | }, |
| 317 | } |
| 318 | |
| 319 | def on_train_end(self): |
| 320 | self.save_model(finish=True) |
nothing calls this directly
no outgoing calls
no test coverage detected