GitHub repository: Arshita27/Pytorch-Model-Memory-Allocation
Primary language: Python
Description: Get total parameters in any created Pytorch model and compare with available memory.