diff --git a/.gitignore b/.gitignore index a83bc02e..459b73bd 100644 --- a/.gitignore +++ b/.gitignore @@ -5,6 +5,7 @@ datasets/* options/* codes/*.txt codes/wandb/* +codes/pretrained_models/* data/* .vscode diff --git a/codes/scripts/audio/random_mp3_splitter.py b/codes/scripts/audio/random_mp3_splitter.py index 01f52638..758bf6d0 100644 --- a/codes/scripts/audio/random_mp3_splitter.py +++ b/codes/scripts/audio/random_mp3_splitter.py @@ -23,7 +23,7 @@ if __name__ == '__main__': separator = Separator('spleeter:2stems') files = find_audio_files(src_dir, include_nonwav=True) for e, file in enumerate(tqdm(files)): - if e < 3055: + if e < 1: continue file_basis = osp.relpath(file, src_dir)\ .replace('/', '_')\ diff --git a/codes/scripts/audio/test_audio_gen.py b/codes/scripts/audio/test_audio_gen.py index 32ab641f..721b2b9b 100644 --- a/codes/scripts/audio/test_audio_gen.py +++ b/codes/scripts/audio/test_audio_gen.py @@ -54,7 +54,7 @@ if __name__ == "__main__": torch.backends.cudnn.benchmark = True want_metrics = False 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.dict_to_nonedict(opt) utils.util.loaded_options = opt