May you consider to update `torchsummary` to `torchinfo`, thanks! - torchsummary - https://github.com/sksq96/pytorch-summary - torchinfo - https://github.com/TylerYep/torchinfo
May you consider to update
torchsummarytotorchinfo, thanks!