7a90d38920857d0155f3cbbfd36d1f5cb91e8eba,models/densenet.py,DenseNet,__init__,#DenseNet#Any#Any#Any#Any#Any#Any#Any#Any#,143

Before Change

                m.weight = nn.init.kaiming_normal(m.weight, mode="fan_out")
            elif isinstance(m, nn.BatchNorm3d) or isinstance(m, nn.BatchNorm2d):
                m.weight.data.fill_(1)
                m.bias.data.zero_()

        // Linear layer
        self.classifier = nn.Linear(num_features, num_classes)

After Change

                    m.weight, mode="fan_out", nonlinearity="relu")
            elif isinstance(m, nn.BatchNorm3d):
                nn.init.constant_(m.weight, 1)
                nn.init.constant_(m.bias, 0)
            elif isinstance(m, nn.Linear):
                nn.init.constant_(m.bias, 0)

    def forward(self, x):
        features = self.features(x)
        out = F.relu(features, inplace=True)

Instances