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

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

@ -142,15 +142,12 @@ 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")) )
) feat = self.lrelu(
self.conv_up2(F.interpolate(feat, scale_factor=2, mode="nearest"))
if self.scale == 4: )
feat = self.lrelu(
self.conv_up2(F.interpolate(feat, scale_factor=2, mode="nearest"))
)
out = self.conv_last(self.lrelu(self.conv_hr(feat))) out = self.conv_last(self.lrelu(self.conv_hr(feat)))