can remove aligner

1 files changed, 3 insertions, 2 deletions

diff --git a/become_yukarin/loss.py b/become_yukarin/loss.py
index c59747a..b2b03fc 100644
--- a/become_yukarin/loss.py
+++ b/become_yukarin/loss.py
@@ -7,7 +7,7 @@ from .model import Predictor
 
 
 class Loss(chainer.link.Chain):
-    def __init__(self, config: LossConfig, predictor: Predictor, aligner: Aligner):
+    def __init__(self, config: LossConfig, predictor: Predictor, aligner: Aligner = None):
         super().__init__()
         self.config = config
 
@@ -21,7 +21,8 @@ class Loss(chainer.link.Chain):
         mask = chainer.as_variable(mask)
 
         h = input
-        h = self.aligner(h)
+        if self.aligner is not None:
+            h = self.aligner(h)
         y = self.predictor(h)
 
         loss = chainer.functions.sum(chainer.functions.absolute_error(y, target) * mask)