apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

remove broken channel adjustment

This commit is contained in:
Sean Sube 2023-12-30 11:59:37 -06:00
parent 477747cced
commit 11e643bcb5
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/models/rrdb.py
View File

@ -88,10 +88,6 @@ class RRDBNet(nn.Module):
): ):
super(RRDBNet, self).__init__() super(RRDBNet, self).__init__()
self.scale = scale 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) 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]) print([num_in_ch, num_out_ch, num_feat, num_block, num_grow_ch, scale])