Update preprocess.py

pull/3905/head
张春乔 10 months ago committed by GitHub
parent 6dd39a291a
commit 85e8c4e903
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -228,9 +228,9 @@ def main():
if args.dataset == "baker": if args.dataset == "baker":
wav_files = sorted(list((rootdir / "Wave").rglob("*.wav"))) wav_files = sorted(list((rootdir / "Wave").rglob("*.wav")))
# split data into 3 sections, the max number of dev/test is 10% or 100 # split data into 3 sections
num_dev = min(int(len(wav_files) * 0.1), 100) num_train = 9800
num_train = len(wav_files) - num_dev * 2 num_dev = 100
train_wav_files = wav_files[:num_train] train_wav_files = wav_files[:num_train]
dev_wav_files = wav_files[num_train:num_train + num_dev] dev_wav_files = wav_files[num_train:num_train + num_dev]
test_wav_files = wav_files[num_train + num_dev:] test_wav_files = wav_files[num_train + num_dev:]

Loading…
Cancel
Save