Update modeling_wavcoch.py
Browse files- modeling_wavcoch.py +1 -1
modeling_wavcoch.py
CHANGED
|
@@ -748,7 +748,7 @@ class WavCoch(PreTrainedModel):
|
|
| 748 |
|
| 749 |
# pad input waveform to correct for cutoff performed by cochleagram
|
| 750 |
if pad:
|
| 751 |
-
wav = F.pad(wav, (
|
| 752 |
|
| 753 |
|
| 754 |
# quantize audio
|
|
|
|
| 748 |
|
| 749 |
# pad input waveform to correct for cutoff performed by cochleagram
|
| 750 |
if pad:
|
| 751 |
+
wav = F.pad(wav, (self.N - self.hop_length, 0), mode='constant', value=0)
|
| 752 |
|
| 753 |
|
| 754 |
# quantize audio
|