diff options
| author | Hiroshiba Kazuyuki <kazuyuki_hiroshiba@dwango.co.jp> | 2018-02-28 05:18:13 +0900 |
|---|---|---|
| committer | Hiroshiba Kazuyuki <kazuyuki_hiroshiba@dwango.co.jp> | 2018-02-28 05:18:13 +0900 |
| commit | b77e381daafa6ec62476a9ccf1e2cdc2460a6902 (patch) | |
| tree | 516f44d314296741622053da2d2b1fc315de687b /become_yukarin/model/sr_model.py | |
| parent | a7606629eab569cf98029870c8414ffaa6d320b7 (diff) | |
超解像が動かないバグを修正
Diffstat (limited to 'become_yukarin/model/sr_model.py')
| -rw-r--r-- | become_yukarin/model/sr_model.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/become_yukarin/model/sr_model.py b/become_yukarin/model/sr_model.py index 12863a7..28ba0c4 100644 --- a/become_yukarin/model/sr_model.py +++ b/become_yukarin/model/sr_model.py @@ -94,8 +94,8 @@ class SRPredictor(chainer.Chain): def __init__(self, in_ch, out_ch, base, extensive_layers) -> None: super().__init__() with self.init_scope(): - self.encoder = Encoder(in_ch, base=base, extensive_layers=extensive_layers) - self.decoder = Decoder(out_ch, base=base, extensive_layers=extensive_layers) + self.encoder = SREncoder(in_ch, base=base, extensive_layers=extensive_layers) + self.decoder = SRDecoder(out_ch, base=base, extensive_layers=extensive_layers) def __call__(self, x): return self.decoder(self.encoder(x)) |