#170 · 主分类： MLOps与评测
pytorch-summary
PyTorch中的模型摘要，类似于Keras中的`model.summary()`
项目最后更新：03/02/24
GitHub Stars
4.1K
Forks数量
409
贡献者数量
11
许可证
MIT
收录理由
用 print(model) 查看 PyTorch 模型结构既费劲，又看不出内存占用。这个工具能打印出类似 Keras 风格的摘要：逐层列出输出形状和参数量，还附带输入尺寸以及前向/反向传播所占内存的估算值。模块之间的形状不匹配一眼就能发现，内存数字也能帮你判断显卡能塞下多大的 batch。作者后来把用户引向了维护更积极的 torchinfo，所以这个仓库更适合看作轻量级的原版，以及快速理解实现思路的参考。对于老一点的 PyTorch 代码，或者你只想用一个函数、不想引入额外依赖的场景，它依然能轻松嵌入。
相关文章
暂无针对该项目的文章。
如需补充新闻资讯、使用教程、应用测评等内容，欢迎 联系我们推荐选题或投稿。
同类热门项目
unsloth
本地UI，用于运行和训练LLM及扩散模型。支持GGUF、MLX、Qwen3.8、Kimi K3、MiniMax-H3、Gemma 4、FLUX等。
LlamaFactory
统一高效微调100+大语言模型与视觉语言模型（ACL 2024）
airflow
Apache Airflow - 一个以编程方式编写、调度和监控工作流的平台
langfuse
🪢 开源AI工程平台：LLM评估、可观测性、指标、提示管理、游乐场、数据集。与OpenTelemetry、LangChain、OpenAI SDK、LiteLLM等集成。🍊YC W23
netron
神经网络、深度学习和机器学习模型的可视化工具