(self, batch_idx, model_out, sample)
| 137 | return losses, output |
| 138 | |
| 139 | def plot_pitch(self, batch_idx, model_out, sample): |
| 140 | gt_f0 = denorm_f0(sample['f0'], sample['uv'], hparams) |
| 141 | self.logger.experiment.add_figure( |
| 142 | f'f0_{batch_idx}', |
| 143 | f0_to_figure(gt_f0[0], None, model_out['f0_denorm_pred'][0]), |
| 144 | self.global_step) |
| 145 | |
| 146 | def add_pitch_loss(self, output, sample, losses): |
| 147 | # mel2ph = sample['mel2ph'] # [B, T_s] |
no test coverage detected