search

sksq96 / pytorch-summary

Model summary in PyTorch similar to `model.summary()` in Keras
MIT License
3.98k stars 412 forks source link

Granular control over information we get from model summary #121

Closed sidwa closed 3 years ago

sidwa commented 4 years ago

The summary function seems to print the model summary. However I'm trying to extract only total size of model from the summary for multiple models. It would be great if we had access to the summary_string() function so that we could extract the info we want using re or the summary() function returned all the parameter it calculated in a dictionary.

EDIT: summary_string() is available but through this github repo. PyPi version has not been updated.

TylerYep commented 4 years ago

Fixed in my fork - see #124