热门
导航
快讯
推荐文章
热门
导航
快讯
推荐文章
#model summary
torchinfo - 高级模型结构查看工具,适用于PyTorch
torchinfo
PyTorch
model summary
Tensorflow
API
Github
开源项目
Torchinfo 提供了类似 TensorFlow `model.summary()` API 的功能,可视化和调试 PyTorch 模型。支持包括 RNN 和 LSTM 在内的多种层,并返回 ModelStatistics 对象。项目拥有简洁界面、多种自定义选项和详细文档,适用于 Jupyter Notebook 和 Google Colab,且经过综合单元测试和代码覆盖测试验证。
1
1
相关文章
torchinfo: PyTorch模型可视化与分析工具
3 个月前
使用协议
隐私政策
广告服务
投诉举报邮箱: service@vectorlightyear.com
@2024 懂AI
·
鲁ICP备2024100362号-6
·
鲁公网安备37021002001498号