diff options
| author | Hiroshiba Kazuyuki <kazuyuki_hiroshiba@dwango.co.jp> | 2017-11-16 06:36:51 +0900 |
|---|---|---|
| committer | Hiroshiba Kazuyuki <kazuyuki_hiroshiba@dwango.co.jp> | 2017-11-16 06:36:51 +0900 |
| commit | 3a69b426bfaed71f9ba1ca02fe0767300ac05a06 (patch) | |
| tree | c3b32d7caa27f0755b2013f19535679261c39ff9 /become_yukarin/config.py | |
| parent | 8520cf40ef62136c5b3e5d4139c97dad50107404 (diff) | |
add gru
Diffstat (limited to 'become_yukarin/config.py')
| -rw-r--r-- | become_yukarin/config.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/become_yukarin/config.py b/become_yukarin/config.py index 424e598..864de4e 100644 --- a/become_yukarin/config.py +++ b/become_yukarin/config.py @@ -16,6 +16,7 @@ class DatasetConfig(NamedTuple): target_mean_path: Path target_var_path: Path features: List[str] + train_crop_size: int seed: int num_test: int @@ -82,6 +83,7 @@ def create_from_json(s: Union[str, Path]): target_mean_path=Path(d['dataset']['target_mean_path']).expanduser(), target_var_path=Path(d['dataset']['target_var_path']).expanduser(), features=d['dataset']['features'], + train_crop_size=d['dataset']['train_crop_size'], seed=d['dataset']['seed'], num_test=d['dataset']['num_test'], ), |