(self)
| 160 | self.scheduler.step(self.global_step // hparams['accumulate_grad_batches']) |
| 161 | |
| 162 | def on_epoch_end(self): |
| 163 | loss_outputs = {k: round(v.avg, 4) for k, v in self.training_losses_meter.items()} |
| 164 | print(f"\n==============\n " |
| 165 | f"Epoch {self.current_epoch} ended. Steps: {self.global_step}. {loss_outputs}" |
| 166 | f"\n==============\n") |
| 167 | |
| 168 | def validation_step(self, sample, batch_idx): |
| 169 | """ |
nothing calls this directly
no outgoing calls
no test coverage detected