torchinfo:PyTorch模型可视化工具，类似Tensorflow的model.summary()

Star1

Star1

View model summaries in PyTorch!

项目介绍

在 PyTorch 中查看模型概要！【此简介由AI生成】

MIT Python 553提交数keras python pytorch torch torch-summary torchinfo torchsummary torchvision visualization

定制我的领域

152.95 K138访问 GitHub