|
|
@@ -108,9 +108,9 @@ class RTCBlock(nn.Module):
|
|
|
super(RTCBlock, self).__init__()
|
|
|
self.inter_dim = out_dim // 2
|
|
|
self.input_proj = BasicConv(in_dim, out_dim, kernel_size=1, act_type=act_type, norm_type=norm_type)
|
|
|
- self.m = nn.Sequential(*(
|
|
|
+ self.m = nn.ModuleList([
|
|
|
Bottleneck(self.inter_dim, self.inter_dim, 1.0, [3, 3], shortcut, act_type, norm_type, depthwise)
|
|
|
- for _ in range(num_blocks)))
|
|
|
+ for _ in range(num_blocks)])
|
|
|
self.output_proj = BasicConv((2 + num_blocks) * self.inter_dim, out_dim, kernel_size=1, act_type=act_type, norm_type=norm_type)
|
|
|
|
|
|
def forward(self, x):
|