文件名称:PyTorch中的模型分析器-Python开发
文件大小:14KB
文件格式:ZIP
更新时间:2024-06-14 09:53:14
Python Deep Learning
torchstat这是基于PyTorch的轻量级神经网络分析仪。 它旨在使您能够快速,轻松地构建网络并进行调试。 注意:此存储库当前正在开发中。 torchstat这是基于PyTorch的轻量级神经网络分析仪。 它旨在使您能够快速,轻松地构建网络并进行调试。 注意:此存储库当前正在开发中。 因此,某些API可能会更改。 该工具可以显示网络参数总数浮点算术(FLOP)的理论数量乘加(MAdd)的理论数量内存使用情况安装有两种方法可以将torchstat安装到您的环境中
【文件预览】:
torchstat-master
----.travis.yml(188B)
----example.py(801B)
----test_requirements.txt(11B)
----detail.md(338B)
----LICENSE(1KB)
----torchstat()
--------reporter.py(3KB)
--------statistics.py(3KB)
--------model_hook.py(4KB)
--------__main__.py(1KB)
--------stat_tree.py(5KB)
--------__init__.py(650B)
--------compute_flops.py(3KB)
--------compute_memory.py(3KB)
--------compute_madd.py(5KB)
----setup.cfg(91B)
----requirements.txt(19B)
----setup.py(1KB)
----.gitignore(1KB)
----README.md(4KB)