diff options
| author | Hiroshiba Kazuyuki <kazuyuki_hiroshiba@dwango.co.jp> | 2017-12-24 20:26:32 +0900 |
|---|---|---|
| committer | Hiroshiba Kazuyuki <kazuyuki_hiroshiba@dwango.co.jp> | 2017-12-24 20:27:25 +0900 |
| commit | 3b38bf420774f2a7f718be927689b67446e680c9 (patch) | |
| tree | 8c18c84042e500e5ff78729a10d21481f7bd4903 /become_yukarin/dataset | |
| parent | 93df4c160b8332a4ef41190860b5056905143def (diff) | |
separate-noise-level
Diffstat (limited to 'become_yukarin/dataset')
| -rw-r--r-- | become_yukarin/dataset/dataset.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/become_yukarin/dataset/dataset.py b/become_yukarin/dataset/dataset.py index fa68a78..5ad7a80 100644 --- a/become_yukarin/dataset/dataset.py +++ b/become_yukarin/dataset/dataset.py @@ -420,11 +420,11 @@ def create(config: DatasetConfig): data_process_train.append(SplitProcess(dict( input=ChainProcess([ LambdaProcess(lambda d, test: d['input']), - AddNoiseProcess(p_global=config.global_noise, p_local=config.local_noise), + AddNoiseProcess(p_global=config.input_global_noise, p_local=config.input_local_noise), ]), target=ChainProcess([ LambdaProcess(lambda d, test: d['target']), - AddNoiseProcess(p_global=config.global_noise, p_local=config.local_noise), + AddNoiseProcess(p_global=config.target_global_noise, p_local=config.target_local_noise), ]), mask=ChainProcess([ LambdaProcess(lambda d, test: d['mask']), |