From 366efd1a584c7ddb96447ccec4703a5ba60c9e21 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 30 Dec 2023 13:34:12 -0600 Subject: [PATCH] fix RRDB rescale arch --- api/onnx_web/models/rrdb.py | 15 ++++++--------- 1 file changed, 6 insertions(+), 9 deletions(-) diff --git a/api/onnx_web/models/rrdb.py b/api/onnx_web/models/rrdb.py index e2c77b42..601767fd 100644 --- a/api/onnx_web/models/rrdb.py +++ b/api/onnx_web/models/rrdb.py @@ -142,15 +142,12 @@ class RRDBNetRescale(nn.Module): trunk = self.conv_body(self.body(feat)) feat = feat + trunk - if self.scale > 1: - feat = self.lrelu( - self.conv_up1(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")) - ) + feat = self.lrelu( + 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")) + ) out = self.conv_last(self.lrelu(self.conv_hr(feat)))