Add the PyTorch profiler #21
Description
We should upgrade our current tensorboard metric tracker to use the PyTorch profiler and any other interesting, potentially useful new feature you consider viable.
{{ message }}
We should upgrade our current tensorboard metric tracker to use the PyTorch profiler and any other interesting, potentially useful new feature you consider viable.