Optimization, Pytorch, Machine Learning, Deep Learning, Tensorboard.
Performance Results In the image below we show the Trace View result of the script above: Trace View of Baseline Model (Captured by Author) As highlighted, the training step takes roughly 9.35 milliseconds, about a third of which are spent on the forward pass of our cross-entropy loss! This is a