de6a2feddaaa87c34784b9319427360ac086be1a,examples/ResNet/imagenet-resnet.py,Model,__init__,#Model#Any#Any#,24

Before Change

        self.mode = mode
        basicblock = preresnet_basicblock if mode == "preact" else resnet_basicblock
        bottleneck = {
            "resnet": resnet_bottleneck,
            "resnext32x4d": resnext_32x4d_bottleneck,
            "preact": preresnet_bottleneck,
            "se": se_resnet_bottleneck}[mode]
        self.num_blocks, self.block_func = {
            18: ([2, 2, 2, 2], basicblock),
            34: ([3, 4, 6, 3], basicblock),

After Change

            101: ([3, 4, 23, 3], bottleneck),
            152: ([3, 8, 36, 3], bottleneck)
        }[depth]
        assert self.block_func is not None, \
            "(mode={}, depth={}) not implemented!".format(mode, depth)

    def get_logits(self, image):
        with argscope([Conv2D, MaxPooling, GlobalAvgPooling, BatchNorm], data_format=self.data_format):
            return resnet_backbone(

Instances