diff options
| author | Hiroshiba Kazuyuki <kazuyuki_hiroshiba@dwango.co.jp> | 2017-12-04 08:02:06 +0900 |
|---|---|---|
| committer | Hiroshiba Kazuyuki <kazuyuki_hiroshiba@dwango.co.jp> | 2017-12-04 08:02:06 +0900 |
| commit | 421d6aab0953a47fff33062118c45d08ca95b00b (patch) | |
| tree | f2fd8b6deae2726794884dbdcaba5334d7570499 /become_yukarin/config.py | |
| parent | 26b794150ac8b6c1abe4fb20b420fd6bf11bc4c6 (diff) | |
add dragan
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 80212b6..a65a72c 100644 --- a/become_yukarin/config.py +++ b/become_yukarin/config.py @@ -49,6 +49,7 @@ class LossConfig(NamedTuple): predictor_fake: float discriminator_true: float discriminator_fake: float + discriminator_grad: float class TrainConfig(NamedTuple): @@ -135,6 +136,7 @@ def create_from_json(s: Union[str, Path]): predictor_fake=d['loss']['predictor_fake'], discriminator_true=d['loss']['discriminator_true'], discriminator_fake=d['loss']['discriminator_fake'], + discriminator_grad=d['loss']['discriminator_grad'], ), train=TrainConfig( batchsize=d['train']['batchsize'], |