Optimization, Pytorch, Machine Learning, Deep Learning, Tensorboard.
take a closer look at the Trace View of the weighted_nll function which takes up the majority of the loss calculation. Trace View of weighted_nll Function (Captured by Author) We can see from the trace that the function is formed from multiple small blocks, each of which is ultimately mapped to an