search

BayesWatch / pytorch-experiments-template

A pytorch based classification experiments template
GNU General Public License v3.0
5 stars 1 forks source link

Add neat model summary #38

Closed AntreasAntoniou closed 3 years ago

AntreasAntoniou commented 3 years ago

Using this package to do it all https://pypi.org/project/pytorch-model-summary/

AntreasAntoniou commented 3 years ago

Done! @jack-willturner