AI备忘录 - Python库 - torchsummary 打印 Pytorch 模型 的评论
torchsummary 库实现类似于 Keras 的 model.summary() 的功能，以便于打印 PyTorch 模型信息.Github - pytorch-summary安装pip ...