fix RRDB rescale arch
This commit is contained in:
parent
6834b716ea
commit
366efd1a58
|
@ -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)))
|
||||
|
||||
|
|
Loading…
Reference in New Issue