How did you use crossentropy without sigmoid after denselayer ?
fsalmasri opened this issue · 1 comments
fsalmasri commented
How did you use crossentropy without sigmoid after denselayer ?
KangBK0120 commented
Hi @fsalmasri
Sorry for the late reply.
In Pytorch, as far as I know, Softmax function is included in the Cross Entropy loss.
Therefore, I didn't add Softmax layer in the model.
See https://discuss.pytorch.org/t/do-i-need-to-use-softmax-before-nn-crossentropyloss/16739