Some minor issues:
You should call the model directly like this model(inputs) instead of model.forward(inputs). However, that shouldn’t be the issue regarding the performance.
It seems your model returns log probabilities. You can directly get the prediction using torch.argmax(outputs, dim=1) instead of using torch.exp.