Apply suggestions from code review

pull/3906/head
张春乔 10 months ago committed by GitHub
parent b9c1e440fd
commit eda7a35b4e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

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

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

Loading…
Cancel
Save