I tried to change it this way without the predict function when DataParallel was used while training.
predictions  = best_model(x_tensor)
It seems to be predicting, but not sure if this is the right way?