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__':
params = {
'mode': 'unsupervised_audio',
'path': ['Z:\\split\\cleaned\\books0'],
'path': ['Z:\\split\\cleaned\\books0', 'Z:\\split\\cleaned\\books2'],
'cache_path': 'E:\\audio\\remote-cache.pth',
'sampling_rate': 22050,
'pad_to_seconds': 5,
'phase': 'train',
'n_workers': 0,
'n_workers': 4,
'batch_size': 16,
'extra_samples': 4,
}

View File

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