1. 程式人生 > >pytorch-loss function

pytorch-loss function

nn.LogSoftmax()

F.log_softmax()

nn.NLLLoss()

nn.NLLLoss2d()

-----------------reference---------------

1.http://blog.csdn.net/zhangxb35/article/details/72464152?utm_source=itdadao&utm_medium=referral

2. https://discuss.pytorch.org/t/build-your-own-loss-function-in-pytorch/235

3. https://discuss.pytorch.org/t/understanding-loss-function-gradients/771/3