Memory

Memory consumption measurement for PyTorch models

See Also