@@ -99,9 +99,9 @@ class IMDN_RTE(nn.Module):
99
99
def __init__ (self , upscale = 2 , in_nc = 3 , nf = 20 , out_nc = 3 ):
100
100
super (IMDN_RTE , self ).__init__ ()
101
101
self .upscale = upscale
102
- self .fea_conv = nn .Sequential (B .conv_layer (in_nc , nf , 3 , bias = True ),
102
+ self .fea_conv = nn .Sequential (B .conv_layer (in_nc , nf , 3 ),
103
103
nn .ReLU (inplace = True ),
104
- B .conv_layer (nf , nf , 3 , stride = 2 ))
104
+ B .conv_layer (nf , nf , 3 , stride = 2 , bias = False ))
105
105
106
106
self .block1 = IMDModule_Large (nf )
107
107
self .block2 = IMDModule_Large (nf )
@@ -110,9 +110,9 @@ def __init__(self, upscale=2, in_nc=3, nf=20, out_nc=3):
110
110
self .block5 = IMDModule_Large (nf )
111
111
self .block6 = IMDModule_Large (nf )
112
112
113
- self .LR_conv = B .conv_layer (nf , nf , 1 )
113
+ self .LR_conv = B .conv_layer (nf , nf , 1 , bias = False )
114
114
115
- self .upsampler = B .pixelshuffle_block (nf , out_nc , upscale_factor = upscale ** 2 , bias = True )
115
+ self .upsampler = B .pixelshuffle_block (nf , out_nc , upscale_factor = upscale ** 2 )
116
116
117
117
def forward (self , input ):
118
118
0 commit comments