4999f9795af7a21ca69adf5f72301b79f719498e,gluoncv/model_zoo/pspnet.py,PSPNet,__init__,#PSPNet#Any#Any#Any#Any#,33

Before Change

            self.head.collect_params().setattr("lr_mult", 10)
            if self.aux:
                self.auxlayer = _FCNHead(1024, nclass, norm_layer=norm_layer, **kwargs)
                self.auxlayer.initialize()
                self.auxlayer.collect_params().setattr("lr_mult", 10)

    def forward(self, x):
        _, _, H, W = x.shape

After Change

            self.head = _PSPHead(nclass, norm_layer=norm_layer, **kwargs)
            self.head.initialize()
            self.head.collect_params().setattr("lr_mult", 10)
            if self.aux:
                self.auxlayer = _FCNHead(1024, nclass, norm_layer=norm_layer, **kwargs)
                self.auxlayer.initialize()
                self.auxlayer.collect_params().setattr("lr_mult", 10)

Instances