(self, sample, batch_idx)
| 79 | return total_loss, loss_output |
| 80 | |
| 81 | def validation_step(self, sample, batch_idx): |
| 82 | outputs = {} |
| 83 | outputs['losses'] = {} |
| 84 | outputs['losses'], model_out = self.run_model(self.model, sample, return_output=True) |
| 85 | outputs['total_loss'] = sum(outputs['losses'].values()) |
| 86 | outputs['nsamples'] = sample['nsamples'] |
| 87 | mel_out = self.model.out2mel(model_out['mel_out']) |
| 88 | outputs = utils.tensors_to_scalars(outputs) |
| 89 | # if sample['mels'].shape[0] == 1: |
| 90 | # self.add_laplace_var(mel_out, sample['mels'], outputs) |
| 91 | if batch_idx < hparams['num_valid_plots']: |
| 92 | self.plot_mel(batch_idx, sample['mels'], mel_out) |
| 93 | self.plot_dur(batch_idx, sample, model_out) |
| 94 | if hparams['use_pitch_embed']: |
| 95 | self.plot_pitch(batch_idx, sample, model_out) |
| 96 | return outputs |
| 97 | |
| 98 | def _validation_end(self, outputs): |
| 99 | all_losses_meter = { |
nothing calls this directly
no test coverage detected