Latest PyPi package doesn't support summary_string

sksq96 / pytorch-summary

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

MIT License

3.98k stars 412 forks source link

Latest PyPi package doesn't support summary_string #115

Open Kostis-S-Z opened 4 years ago

Kostis-S-Z commented 4 years ago

Thank you for a great package!

I really want to use the summary_string option for my project and installing the package through git works fine (pip install git+https://github.com/sksq96/pytorch-summary.git). But the PyPI version is still behind (pip install torchsummary).

Could you please deploy the latest github version to PyPI?

TylerYep commented 4 years ago

Available in my fork - See #124 It's available via the ModelStatistics object, which can be printed via str(model_stats)

Kostis-S-Z commented 4 years ago

Cool! I will check it out, thanks!

gui-miotto commented 3 years ago

Great job, @TylerYep !