Pytorch, Optimization, Deep Learning, Artificial Intelligence, Tensorboard.
training-step. The advantage to using torch.profiler.record_function labels was that it allowed us to easily zone in on the problematic portions of our model. Ideally, we would like to be able to retain the same capability even in the case of performance issues in the backward pass. In the next