Mukesh1729:

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.


Yes, you can pass the input to the model and use the output as the regression prediction. Since you would usually not update any parameters during evaluation, you could also wrap the forward pass into a with torch.no_grad() guard to save memory.