Thanks,
I am working on a multi-label classification problem, there are 4 classes in total.
And I changed nn.NLLLoss to nn.CrossEntropyLoss and my model is training 
Should I use Adam or SGD optimizers.
Thanks a lot !!