Update torchsummary to torchinfo #18
Description
May you consider to update torchsummary to torchinfo, thanks!
- torchsummary - https://github.com/sksq96/pytorch-summary
- torchinfo - https://github.com/TylerYep/torchinfo
{{ message }}
May you consider to update torchsummary to torchinfo, thanks!