pytorch-memlab (pypi): A lab to do simple and accurate memory experiments on pytorch
Keywords: pytorch|memory|profile|cuda-memory|memory-profiler