Optimization, Pytorch, Machine Learning, Deep Learning, Tensorboard.
caused a slight increase in the time of the forward pass of the loss function. It appears that the CrossEntropyLoss is not meaty enough to benefit from this optimization. You may be wondering why we can’t just apply torch compilation to our initial loss function and rely on it to compile our code