diff options
Diffstat (limited to 'become_yukarin/config.py')
| -rw-r--r-- | become_yukarin/config.py | 10 |
1 files changed, 4 insertions, 6 deletions
diff --git a/become_yukarin/config.py b/become_yukarin/config.py index f74c83e..424e598 100644 --- a/become_yukarin/config.py +++ b/become_yukarin/config.py @@ -9,8 +9,8 @@ from .param import Param class DatasetConfig(NamedTuple): param: Param - input_glob: str - target_glob: str + input_glob: Path + target_glob: Path input_mean_path: Path input_var_path: Path target_mean_path: Path @@ -40,7 +40,6 @@ class TrainConfig(NamedTuple): gpu: int log_iteration: int snapshot_iteration: int - output: Path class Config(NamedTuple): @@ -76,8 +75,8 @@ def create_from_json(s: Union[str, Path]): return Config( dataset=DatasetConfig( param=Param(), - input_glob=d['dataset']['input_glob'], - target_glob=d['dataset']['target_glob'], + input_glob=Path(d['dataset']['input_glob']).expanduser(), + target_glob=Path(d['dataset']['target_glob']).expanduser(), input_mean_path=Path(d['dataset']['input_mean_path']).expanduser(), input_var_path=Path(d['dataset']['input_var_path']).expanduser(), target_mean_path=Path(d['dataset']['target_mean_path']).expanduser(), @@ -104,6 +103,5 @@ def create_from_json(s: Union[str, Path]): gpu=d['train']['gpu'], log_iteration=d['train']['log_iteration'], snapshot_iteration=d['train']['snapshot_iteration'], - output=Path(d['train']['output']).expanduser(), ), ) |