(self, model, sample, return_output=False, infer=False)
| 126 | return outputs |
| 127 | |
| 128 | def run_model(self, model, sample, return_output=False, infer=False): |
| 129 | f0 = sample['f0'] |
| 130 | uv = sample['uv'] |
| 131 | output = model(sample['mels']) |
| 132 | losses = {} |
| 133 | self.add_pitch_loss(output, sample, losses) |
| 134 | if not return_output: |
| 135 | return losses |
| 136 | else: |
| 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) |
no test coverage detected