diff options
Diffstat (limited to 'scripts/process.py')
| -rw-r--r-- | scripts/process.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/scripts/process.py b/scripts/process.py index 8f5c2f0..3a2b788 100644 --- a/scripts/process.py +++ b/scripts/process.py @@ -10,7 +10,8 @@ import numpy from become_yukarin import AcousticConverter from become_yukarin import SuperResolution -from become_yukarin.config.config import create_from_json as create_config +from become_yukarin.config.config import create_from_json as create_conv_config +from become_yukarin.config.sr_config import create_from_json as create_sr_config parser = argparse.ArgumentParser() parser.add_argument('-cmd', '--conv_model_directory', type=Path, default=Path('/mnt/dwango/hiroshiba/become-yukarin/')) @@ -33,7 +34,7 @@ def extract_number(f): return int(s[-1]) if s else -1 def load_acoustic_converter(): - conv_config = create_config(conv_model / 'config.json') + conv_config = create_conv_config(conv_model / 'config.json') if it is not None: conv_model_path = conv_model / 'predictor_{}.npz'.format(it) else: @@ -43,7 +44,7 @@ def load_acoustic_converter(): return AcousticConverter(conv_config, conv_model_path, gpu=gpu) def load_super_resolution(): - sr_config = create_config(sr_model / 'config.json') + sr_config = create_sr_config(sr_model / 'config.json') sr_model_paths = sr_model.glob('predictor*.npz') sr_model_path = list(sorted(sr_model_paths, key=extract_number))[-1] print(sr_model_path) |