6da69d81e7aa7aac21a8a4abf5b13d2caed3dfa9,model.py,ft_net,forward,#ft_net#,64

Before Change

        x = self.model.layer3(x)
        x = self.model.layer4(x)
        x = self.model.avgpool(x)
        x = torch.squeeze(x)
        x = self.classifier(x)
        return x

// Define the DenseNet121-based Model

After Change

        x = self.model.layer3(x)
        x = self.model.layer4(x)
        x = self.model.avgpool(x)
        x = x.view(x.size(0), x.size(1))
        x = self.classifier(x)
        return x

// Define the DenseNet121-based Model

Instances