文件名称：PyTorch中的模型摘要类似于Keras中的`model.summary（）`-Python开发
文件大小：7KB
文件格式：ZIP
更新时间：2024-06-14 06:16:01
Python Deep Learning
PyTorch中的Keras样式model.summary（）Keras有一个简洁的API可以查看模型的可视化，这在调试网络时非常有帮助。 这是一个尝试在PyTorch中模仿的准系统代码。 目的是建立Keras风格的model.PyTorch中的summary（）Keras拥有一个简洁的API来查看模型的可视化，这在调试网络时非常有帮助。 这是一个尝试在PyTorch中模仿的准系统代码。 目的是提供与PyTorch中的print（your_model）无法提供的信息互补的信息。 用法pip install torchsummary或git clone https://github.com/sksq96/pytorch-summary from torchsummary导入摘要摘要（您的模型，input_size =（channels，H，W））注意
【文件预览】：
pytorch-summary-master
----setup.py(347B)
----.gitignore(26B)
----LICENSE(1KB)
----README.md(7KB)
----torchsummary()
--------__init__.py(50B)
--------tests()
--------torchsummary.py(5KB)