3861825719b8e6ef042799cdb7861a72e11eb3bb,pytorch/pytorchcv/models/others/oth_resattnet4.py,AttentionModule_stage1,__init__,#AttentionModule_stage1#Any#Any#,443

Before Change

        down_seq.add_module("down3", nn.Sequential(
            nn.MaxPool2d(kernel_size=3, stride=2, padding=1),
            nn.Sequential(
                ResBlock(in_channels, out_channels),
                ResBlock(in_channels, out_channels)
            )
        ))

After Change

        down_seq = nn.Sequential()
        up_seq = nn.Sequential()
        skip_seq = nn.Sequential()
        for i in range(hourglass_depth):
            down_seq.add_module("down{}".format(i + 1), nn.Sequential(
                nn.MaxPool2d(kernel_size=3, stride=2, padding=1),
                ResBlock(in_channels, out_channels)
            ))
            up_seq.add_module("up{}".format(i + 1), nn.Sequential(
                ResBlock(in_channels, out_channels),
                InterpolationBlock2(scale_factor)))
            if i == 0:
                skip_seq.add_module("skip1", nn.Sequential(
                    ResBlock(in_channels, out_channels),
                    ResBlock(in_channels, out_channels)))
            else:
                skip_seq.add_module("skip{}".format(i + 1), DoubleSkipBlock(in_channels, out_channels))
        self.hg = Hourglass(
            down_seq=down_seq,
            up_seq=up_seq,
            skip_seq=skip_seq,

Instances