/pytorch_modelsize

Estimates the size of a PyTorch model in memory

Primary LanguageJupyter NotebookMIT LicenseMIT

Issues