From 421d6aab0953a47fff33062118c45d08ca95b00b Mon Sep 17 00:00:00 2001
From: Hiroshiba Kazuyuki <kazuyuki_hiroshiba@dwango.co.jp>
Date: Mon, 4 Dec 2017 08:02:06 +0900
Subject: add dragan

---
 become_yukarin/config.py | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'become_yukarin/config.py')

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'],
-- 
cgit v1.2.3-70-g09d2