Update nasnet_mobile.py

Remove softmax from logits

Author: vfdev-5

pretrainedmodels/models/nasnet_mobile.py

@@ -607,7 +607,6 @@ def logits(self, features):
         x = x.view(x.size(0), -1)
         x = self.dropout(x)
         x = self.last_linear(x)
-        x = nn.Softmax(1)(x)
         return x
 
     def forward(self, input):
@@ -659,4 +658,4 @@ def nasnetamobile(num_classes=1001, pretrained='imagenet'):
     input = Variable(torch.randn(2, 3, 224, 224))
     output = model(input)
 
-    print(output.size())
+    print(output.size())