diff options
Diffstat (limited to 'become_yukarin/model/sr_model.py')
| -rw-r--r-- | become_yukarin/model/sr_model.py | 6 |
1 files changed, 5 insertions, 1 deletions
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 |