search

WuJie1010 / Facial-Expression-Recognition.Pytorch

A CNN based pytorch implementation on facial expression recognition (FER2013 and CK+), achieving 73.112% (state-of-the-art) in FER2013 and 94.64% in CK+ dataset
MIT License
1.78k stars 545 forks source link

nn.Linear #80

Closed daredevil6776 closed 4 years ago

daredevil6776 commented 4 years ago

How can we assure that the predicted probabilities of all the classes sums to one using nn.Linear?

WuJie1010 commented 4 years ago

softmax helps to finish it

daredevil6776 commented 4 years ago

softmax helps to finish it But in training or testing there is no softmax layer specified

WuJie1010 commented 4 years ago

softmax helps to finish it But in training or testing there is no softmax layer specified

softmax operation is in the cross-entropy loss function