82d70085e0ad4a355acfd486884a3dc28af1b6b9,chainer_/models/squeezenet.py,SqueezeNet,__init__,#SqueezeNet#Any#Any#Any#Any#Any#Any#,157

Before Change

            in_channels = init_block_channels
            for i, channels_per_stage in enumerate(channels):
                //stage = Sequential()
                self.features.append(partial(
                    F.max_pooling_2d,
                    ksize=3,
                    stride=2))
                for j, out_channels in enumerate(channels_per_stage):
                    expand_channels = out_channels // 2
                    squeeze_channels = out_channels // 8
                    self.features.append(FireUnit(

After Change

                    out_channels=init_block_channels,
                    ksize=init_block_kernel_size))
                in_channels = init_block_channels
                for i, channels_per_stage in enumerate(channels):
                    stage = SimpleSequential()
                    with stage.init_scope():
                        setattr(stage, "pool{}".format(i + 1), partial(
                            F.max_pooling_2d,
                            ksize=3,
                            stride=2))
                        for j, out_channels in enumerate(channels_per_stage):
                            expand_channels = out_channels // 2
                            squeeze_channels = out_channels // 8
                            setattr(stage, "unit{}".format(j + 1), FireUnit(
                                in_channels=in_channels,
                                squeeze_channels=squeeze_channels,
                                expand1x1_channels=expand_channels,
                                expand3x3_channels=expand_channels,
                                residual=((residuals is not None) and (residuals[i][j] == 1))))
                            in_channels = out_channels
                    setattr(self.features, "stage{}".format(i + 1), stage)
                setattr(self.features, "dropout", partial(
                    F.dropout,
                    ratio=0.5))

Instances