Questions about class LabelEncodingFunction(nn.Module): in label_encoding_function.py

[ShirleyHe2020]

Is it necessary to set 5 stages in this function? I mean if 3, 4, 6 stages can work

class LabelEncodingFunction(nn.Module): def init(self, in_channels): super(LabelEncodingFunction, self).init() self.stage1 = nn.Conv2d(in_channels, 128, 7, 2, 3) self.stage2 = Bottleneck(128, 64, 256, 1, False, 2, 1) self.stage3 = nn.Sequential( Bottleneck(256, 128, 512, 1, False, 2, 1), Bottleneck(512, 128, 512, 1, False, 1, 1), ) self.stage4 = nn.Sequential( Bottleneck(512, 256, 1024, 1, False, 2, 1), Bottleneck(1024, 256, 1024, 1, False, 1, 1), ) self.stage5 = Bottleneck(1024, 512, 2048, 1, False, 2, 1)

def forward(self, x):
    x = self.stage1(x)
    outs = []
    for m in [self.stage2, self.stage3, self.stage4, self.stage5]:
        x = m(x)
        outs.append(x)
    return outs