Skip to content

Commit 51383ef

Browse files
committed
fixe dataloader build bug when fine-tuning v1 model
1 parent d334fb7 commit 51383ef

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

data/ft_dataset.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -33,9 +33,9 @@ def __init__(
3333
self.sr = sr
3434
self.mel_fn_args = {
3535
"n_fft": spect_params['n_fft'],
36-
"win_size": spect_params.get('win_length', spect_params['win_size']),
37-
"hop_size": spect_params.get('hop_length', spect_params['hop_size']),
38-
"num_mels": spect_params.get('n_mels', spect_params['num_mels']),
36+
"win_size": spect_params.get('win_length', spect_params.get('win_size', 1024)),
37+
"hop_size": spect_params.get('hop_length', spect_params.get('hop_size', 256)),
38+
"num_mels": spect_params.get('n_mels', spect_params.get('num_mels', 80)),
3939
"sampling_rate": sr,
4040
"fmin": spect_params['fmin'],
4141
"fmax": None if spect_params['fmax'] == "None" else spect_params['fmax'],

0 commit comments

Comments
 (0)