This commit is contained in:
James Betker 2021-08-13 13:58:59 -06:00
parent fff1a59e08
commit 81e91c99de
3 changed files with 3 additions and 2 deletions

1
.gitignore vendored
View File

@ -5,6 +5,7 @@ datasets/*
options/* options/*
codes/*.txt codes/*.txt
codes/wandb/* codes/wandb/*
codes/pretrained_models/*
data/* data/*
.vscode .vscode

View File

@ -23,7 +23,7 @@ if __name__ == '__main__':
separator = Separator('spleeter:2stems') separator = Separator('spleeter:2stems')
files = find_audio_files(src_dir, include_nonwav=True) files = find_audio_files(src_dir, include_nonwav=True)
for e, file in enumerate(tqdm(files)): for e, file in enumerate(tqdm(files)):
if e < 3055: if e < 1:
continue continue
file_basis = osp.relpath(file, src_dir)\ file_basis = osp.relpath(file, src_dir)\
.replace('/', '_')\ .replace('/', '_')\

View File

@ -54,7 +54,7 @@ if __name__ == "__main__":
torch.backends.cudnn.benchmark = True torch.backends.cudnn.benchmark = True
want_metrics = False want_metrics = False
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('-opt', type=str, help='Path to options YAML file.', default='../options/test_gpt_tts_lj.yml') parser.add_argument('-opt', type=str, help='Path to options YAML file.', default='../options/test_lrdvae_audio_mozcv.yml')
opt = option.parse(parser.parse_args().opt, is_train=False) opt = option.parse(parser.parse_args().opt, is_train=False)
opt = option.dict_to_nonedict(opt) opt = option.dict_to_nonedict(opt)
utils.util.loaded_options = opt utils.util.loaded_options = opt