I have a model that trained on to predict one of the 14 class,
But when I go for prediction for a single image,
imageData = Image.open(pathImageFile).convert('RGB')
imageData = self.transformSequence(imageData)
imageData = imageData.unsqueeze_(0)
input = torch.autograd.Variable(imageData)
self.model.cuda()
output = self.model(input.cuda())


The output shape is [1,1024,7,7].
Very very weird than Keras predict. how can I know the output label???