torch-scope (pypi): A Toolkit for Training, Tracking and Saving PyTorch Models
Keywords: gpu-memory|logger|pytorch|tensorboard|toolkit