torchinfo

torchsummaryよりtorchinfoがいいよという話 - Qiita
たまに使うのに torchinfo の備忘録を残してなかった。PyTorch で summary を叩きたい時用。