Fix a bug

ShiZican · Aug 5, 2020 · cdc0826 · cdc0826
1 parent f56f003
commit cdc0826
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/lib/modeling/FPT.py b/lib/modeling/FPT.py
@@ -45,6 +45,7 @@ def group_norm(num_channels):
         self.rt_p4_p2 = RenderTrans(channels_high=feature_dim, channels_low=feature_dim, upsample=False)
         self.rt_p3_p2 = RenderTrans(channels_high=feature_dim, channels_low=feature_dim, upsample=False)
         drop_block = DropBlock2D(block_size=3, drop_prob=0.2)
+
         if with_norm != 'none':
             self.fpn_p5_1x1 = nn.Sequential(*[nn.Conv2d(2048, feature_dim, 1, bias=False), norm(feature_dim)])
             self.fpn_p4_1x1 = nn.Sequential(*[nn.Conv2d(1024, feature_dim, 1, bias=False), norm(feature_dim)])