apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix RRDB rescale arch

This commit is contained in:
Sean Sube 2023-12-30 13:34:12 -06:00
parent 6834b716ea
commit 366efd1a58
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 6 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -142,12 +142,9 @@ class RRDBNetRescale(nn.Module):
trunk = self.conv_body(self.body(feat)) trunk = self.conv_body(self.body(feat))
feat = feat + trunk feat = feat + trunk
if self.scale > 1:
feat = self.lrelu( feat = self.lrelu(
self.conv_up1(F.interpolate(feat, scale_factor=2, mode="nearest")) self.conv_up1(F.interpolate(feat, scale_factor=2, mode="nearest"))
) )
if self.scale == 4:
feat = self.lrelu( feat = self.lrelu(
self.conv_up2(F.interpolate(feat, scale_factor=2, mode="nearest")) self.conv_up2(F.interpolate(feat, scale_factor=2, mode="nearest"))
) )