Return model size info - Githubissues

sksq96 / pytorch-summary

Model summary in PyTorch similar to `model.summary()` in Keras

MIT License

3.98k stars 412 forks source link

Open mhamdan91 opened 2 years ago

mhamdan91 commented 2 years ago

This is a useful information to dynamically compute max/recommended batch size.