7ff331f871b563d1dbb36472de28d1a692d8b4af,torchreid/models/mobilenetv2.py,MobileNetV2,forward,#MobileNetV2#Any#,113

Before Change

    def forward(self, x):
        x = self.featuremaps(x)
        x = F.avg_pool2d(x, x.size()[2:]).view(x.size(0), -1)
        x = F.dropout(x, training=self.training)

        if not self.training:
            return x

After Change

    def forward(self, x):
        f = self.featuremaps(x)
        v = self.global_avgpool(f)
        v = v.view(v.size(0), -1)
        
        if self.fc is not None:
            v = self.fc(v)

        if not self.training:
            return v
        
        y = self.classifier(v)

Instances