diff options
Diffstat (limited to 'become_yukarin/config/sr_config.py')
| -rw-r--r-- | become_yukarin/config/sr_config.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/become_yukarin/config/sr_config.py b/become_yukarin/config/sr_config.py index 4c62808..266ea04 100644 --- a/become_yukarin/config/sr_config.py +++ b/become_yukarin/config/sr_config.py @@ -12,6 +12,8 @@ class SRDatasetConfig(NamedTuple): param: Param input_glob: Path train_crop_size: int + input_global_noise: float + input_local_noise: float seed: int num_test: int @@ -75,6 +77,8 @@ def create_from_json(s: Union[str, Path]): param=Param(), input_glob=Path(d['dataset']['input_glob']), train_crop_size=d['dataset']['train_crop_size'], + input_global_noise=d['dataset']['input_global_noise'], + input_local_noise=d['dataset']['input_local_noise'], seed=d['dataset']['seed'], num_test=d['dataset']['num_test'], ), |