CrossEntropyLoss (内置了Softmax) 不要再用softmax
计算公式如下:x为一个num_class(类别数量)维的向量,class为样本的类别。
参考
1. https://blog.csdn.net/geter_CS/article/details/84857220
2. https://blog.csdn.net/Tianweidadada/article/details/82630735
计算公式如下:x为一个num_class(类别数量)维的向量,class为样本的类别。
1. https://blog.csdn.net/geter_CS/article/details/84857220
2. https://blog.csdn.net/Tianweidadada/article/details/82630735