Wiki / Lesson Thread: Lesson 8
About the Intro to Machine Learning category
I have a few questions from the class –
net = nn.Sequential( nn.Linear(28*28, 10), nn.LogSoftmax() )
In last non-linear layer, why did we use
softmax? Weren’t we exponentiating outputs from 2nd last layer so as to make them all +ve ? Why back to
log after doing [exp]/[sum of exp].
What is the reason of dividing by dims. I tried and it doesn’t work if we don’t divide by dims. By it doesn’t work, I mean fit() gives loss = nans and very bad accuracy.
I just posted the video.
The loss functions in pytorch generally assume you have LogSoftmax, for computational efficiency reasons: https://discuss.pytorch.org/t/does-nllloss-handle-log-softmax-and-softmax-in-the-same-way/8835
This is He initialization (http://www.jefkine.com/deep/2016/08/08/initialization-of-deep-networks-case-of-rectifiers/) . Although I may have forgotten a
Without careful initialization you’ll get gradient explosion. We discuss this in the DL course.
Helpful links. Thanks