Closes #222 (closed)
pip install torch_tb_profiler tensorboard --logdir=./log
http://localhost:6006/#pytorch_profiler