From c0f3eacabde5d41992a5ae1d8d8f0f170f6b155e Mon Sep 17 00:00:00 2001
From: Hiroshiba Kazuyuki <hihokaruta@gmail.com>
Date: Mon, 15 Jan 2018 08:24:27 +0900
Subject: 超解像クラス
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 become_yukarin/model/sr_model.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

(limited to 'become_yukarin/model/sr_model.py')

diff --git a/become_yukarin/model/sr_model.py b/become_yukarin/model/sr_model.py
index 64158ca..8f9a34b 100644
--- a/become_yukarin/model/sr_model.py
+++ b/become_yukarin/model/sr_model.py
@@ -111,7 +111,11 @@ class SRDiscriminator(chainer.Chain):
         return h
 
 
+def create_predictor_sr(config: SRModelConfig):
+    return SRPredictor(in_ch=1, out_ch=1)
+
+
 def create_sr(config: SRModelConfig):
-    predictor = SRPredictor(in_ch=1, out_ch=1)
+    predictor = create_predictor_sr()
     discriminator = SRDiscriminator(in_ch=1, out_ch=1)
     return predictor, discriminator
-- 
cgit v1.2.3-70-g09d2