diff options
Diffstat (limited to 'become_yukarin/dataset/dataset.py')
| -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']), |