Estimates the size of a PyTorch model in memory - View it on GitHub