Hey,
I tried helping you in other thread which you started for the same problem ?All the suggestions seemed to be not much helpful from what you replied back. Could you please summarize a few which helped ?
What did you end up using CrossEntropyLoss() ? Did you tweak hyper parameters more ?