Optimization, Pytorch, Machine Learning, Deep Learning, Tensorboard.
in on the loss function, we can see that a significant amount of time is now spent on initializing the torch.nn.NLLLoss object! Trace View of the Loss Function Following Optimization #4 (Captured by Author) Looking back at our loss function, we can see that we are initializing a new NLLLoss object