Regression Not Working #6949
Unanswered
tahsin314
asked this question in
code help: CV
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
I'm working on a regression task. My model is supposed to take an image of retina as input and predict the severity level of an abnormality. The severity level can be any integer between 0 and 4. Here is my lightning module. The code works perfectly if I use
ordinal regression
but does not converge when I useregression
andMSELoss
. (Train Loss and Validation Loss). I've tried varying learning rate from10^-5
to10^1
but there isn't any improvement. I've also tried to overfit my model but it didn't help either. It would be a great help if someone can look into it. Thanks.Beta Was this translation helpful? Give feedback.
All reactions