From 11e643bcb531d6288c4af963e1761fbf553b5f2b Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 30 Dec 2023 11:59:37 -0600 Subject: [PATCH] remove broken channel adjustment --- api/onnx_web/models/rrdb.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/api/onnx_web/models/rrdb.py b/api/onnx_web/models/rrdb.py index 54a43e2f..3f067260 100644 --- a/api/onnx_web/models/rrdb.py +++ b/api/onnx_web/models/rrdb.py @@ -88,10 +88,6 @@ class RRDBNet(nn.Module): ): super(RRDBNet, self).__init__() self.scale = scale - if scale == 2: - num_in_ch = num_in_ch * 4 - elif scale == 1: - num_in_ch = num_in_ch * 16 RRDB_block_f = functools.partial(RRDB, nf=num_feat, gc=num_grow_ch) print([num_in_ch, num_out_ch, num_feat, num_block, num_grow_ch, scale])