can train

1 files changed, 24 insertions, 0 deletions

diff --git a/become_yukarin/loss.py b/become_yukarin/loss.py
new file mode 100644
index 0000000..c088691
--- /dev/null
+++ b/become_yukarin/loss.py
@@ -0,0 +1,24 @@
+from .config import LossConfig
+from .model import Model
+
+import chainer
+
+from chainer import reporter
+
+
+class Loss(chainer.link.Chain):
+    def __init__(self, config: LossConfig, predictor: Model):
+        super().__init__()
+        self.config = config
+
+        with self.init_scope():
+            self.predictor = predictor
+
+    def __call__(self, input, target):
+        h = input
+        y = self.predictor(h)
+
+        loss = chainer.functions.mean_absolute_error(y, target)
+        reporter.report({'loss': loss}, self)
+
+        return loss * self.config.l1