文件名称：torchinfo:在PyTorch中查看模型摘要！
文件大小：36KB
文件格式：ZIP
更新时间：2024-04-09 07:35:10
visualization keras torch pytorch torchvision
火炬信息 （以前是火炬摘要） Torchinfo提供的信息与PyTorch中的print(your_model)提供的信息类似，类似于Tensorflow的model.summary() API，用于查看模型的可视化，这在调试网络时非常有用。在此项目中，我们在PyTorch中实现了类似的功能，并创建了一个干净，简单的界面以在您的项目中使用。 这是@ sksq96和@nmhkahn对原始torchsummary和torchsummaryX项目的完全重写版本。该项目解决了所有问题，并通过引入全新的API提出了原始项目上的剩余请求。 用法 pip install torchinfo 如何使用 from torchinfo import summary model = ConvNet () batch_size = 16 summary ( model , input_size = ( batc
【文件预览】：
torchinfo-main
----setup.py(55B)
----.gitignore(2KB)
----conftest.py(359B)
----codecov.yml(129B)
----requirements.txt(18B)
----.travis.yml(267B)
----.pre-commit-config.yaml(980B)
----LICENSE(1KB)
----setup.cfg(1KB)
----torchinfo()
--------__init__.py(137B)
--------model_statistics.py(6KB)
--------layer_info.py(6KB)
--------formatting.py(2KB)
--------torchinfo.py(14KB)
----.deepsource.toml(167B)
----README.md(16KB)
----tests()
--------torchinfo_test.py(8KB)
--------output_test.py(5KB)
--------__init__.py(0B)
--------exceptions_test.py(2KB)
--------test_output()
--------jupyter_test.ipynb(7KB)
----fixtures()
--------__init__.py(0B)
--------models.py(12KB)
----requirements-dev.txt(57B)