This commit is contained in:
lyuxiang.lx
2026-01-05 11:01:19 +08:00
parent 46788c7379
commit 1ceb2b7b1e

View File

@@ -146,7 +146,7 @@ def Dataset(data_list_file,
partition=partition) partition=partition)
# map partial arg to padding func # map partial arg to padding func
for i in range(1, len(data_pipeline)): for i in range(1, len(data_pipeline)):
if data_pipeline[i].func.__name__ == 'compute_fbank': if data_pipeline[i].func.__name__ == 'compute_fbank' and gan is True:
data_pipeline[i] = partial(data_pipeline[i], token_mel_ratio=0) data_pipeline[i] = partial(data_pipeline[i], token_mel_ratio=0)
if data_pipeline[i].func.__name__ == 'padding': if data_pipeline[i].func.__name__ == 'padding':
data_pipeline[i] = partial(data_pipeline[i], gan=gan, dpo=dpo) data_pipeline[i] = partial(data_pipeline[i], gan=gan, dpo=dpo)