Merge pull request #430 from sramakrishnan247/patch-1

Update mudeep.py to support testing (return features if self.training=False)

torchreid/models/mudeep.py

@@ -195,6 +195,9 @@ class MuDeep(nn.Module):
         x = self.fc(x)
         y = self.classifier(x)
         
+        if not self.training:
+            return x
+        
         if self.loss == 'softmax':
             return y
         elif self.loss == 'triplet':