pytorch-summary
pytorch-summary copied to clipboard
Return model size info
This is a useful information to dynamically compute max/recommended batch size.