Thanks @KFrank,
I appreciate your feedback.
Can you recommend me any tutorial where Pytorch is used for Logistic Regression where the input is different from an image?
All tutorials I find use MNIST dataset.
I try to use Pytorch for a normal mulit-class classification problem, where I have over 20 variables and 1 target variable (5 class)
Thanks again for your time.