From 9f87a74de09e38f9d8f3e7ebb5fd26fac44a3b0e Mon Sep 17 00:00:00 2001 From: Hiroshiba Kazuyuki Date: Wed, 22 Nov 2017 23:50:31 +0900 Subject: can remove aligner --- become_yukarin/loss.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'become_yukarin/loss.py') 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) -- cgit v1.2.3-70-g09d2