fix bug

Author: yxlllc

training/nsf_HiFigan_task.py

@@ -91,13 +91,11 @@ def __getitem__(self, index):
                 start = random.randint(0, audio.shape[0] - 1 - crop_wav_samples)
                 end = start + crop_wav_samples
                 audio = audio[start:end]
-                f0, uv = get_pitch(audio.numpy(), hparams=self.config, speed=speed, interp_uv=True, length=mel.shape[-1])
-                if f0 is None:
-                    return {'f0': data['f0'], 'spectrogram': data['mel'], 'audio': data['audio']}
                 audio_aug = wav_aug(audio, self.config["hop_size"], speed=speed)
                 mel_aug = dynamic_range_compression_torch(self.mel_spec_transform(audio_aug[None,:]))
-                audio_aug = audio_aug[2*samples_per_frame: -2*samples_per_frame].numpy()
-                mel_aug = mel_aug[0, :, 2:-2].T.numpy()
+                f0, uv = get_pitch(audio.numpy(), hparams=self.config, speed=speed, interp_uv=True, length=mel_aug.shape[-1])
+                if f0 is None:
+                    return {'f0': data['f0'], 'spectrogram': data['mel'], 'audio': data['audio']}
                 f0_aug = f0[2:-2] * speed
                 return {'f0': f0_aug, 'spectrogram': mel_aug, 'audio': audio_aug}