Hey thanks for getting back. If I am doing inference on an image for a regression problem, is it similar to doing with a classification problem or just pass the image though the model and get the outputs.
I am using the mse loss function
Ex: train a classifier example from pytorch on cifar