AI备忘录 - Python库 - torchsummary 打印 Pytorch 模型 的评论
type: channel
link: https://www.aiuai.cn/aifarm458.html
description: torchsummary 库实现类似于 Keras 的 model.summary() 的功能，以便于打印 PyTorch 模型信息.Github - pytorch-summary安装pip ...