This commit is contained in:
James Betker 2021-09-16 10:53:56 -06:00
parent 6f48674647
commit 1197ae1928
2 changed files with 3 additions and 3 deletions

View File

@ -136,12 +136,12 @@ class UnsupervisedAudioDataset(torch.utils.data.Dataset):
if __name__ == '__main__': if __name__ == '__main__':
params = { params = {
'mode': 'unsupervised_audio', 'mode': 'unsupervised_audio',
'path': ['Z:\\split\\cleaned\\books0'], 'path': ['Z:\\split\\cleaned\\books0', 'Z:\\split\\cleaned\\books2'],
'cache_path': 'E:\\audio\\remote-cache.pth', 'cache_path': 'E:\\audio\\remote-cache.pth',
'sampling_rate': 22050, 'sampling_rate': 22050,
'pad_to_seconds': 5, 'pad_to_seconds': 5,
'phase': 'train', 'phase': 'train',
'n_workers': 0, 'n_workers': 4,
'batch_size': 16, 'batch_size': 16,
'extra_samples': 4, 'extra_samples': 4,
} }

View File

@ -19,7 +19,7 @@ def main():
maximum_duration = 20 maximum_duration = 20
files = find_audio_files(args.path, include_nonwav=True) files = find_audio_files(args.path, include_nonwav=True)
for e, wav_file in enumerate(tqdm(files)): for e, wav_file in enumerate(tqdm(files)):
#if e < 1326: #if e < 1459:
# continue # continue
print(f"Processing {wav_file}..") print(f"Processing {wav_file}..")
outdir = os.path.join(args.out, f'{e}_{os.path.basename(wav_file[:-4])}').replace('.', '').strip() outdir = os.path.join(args.out, f'{e}_{os.path.basename(wav_file[:-4])}').replace('.', '').strip()