pytorch-summary:PyTorch中的模型摘要类似于Keras中的`model.summary()`。

时间:2024-02-26 03:27:44
【文件属性】:

文件名称:pytorch-summary:PyTorch中的模型摘要类似于Keras中的`model.summary()`。

文件大小:7KB

文件格式:ZIP

更新时间:2024-02-26 03:27:44

deep-learning keras pytorch summary DeeplearningPython

model.summary()中的Keras样式model.summary() Keras有一个简洁的API来查看模型的可视化,这在调试网络时非常有帮助。 这是一个尝试在PyTorch中模仿的准系统代码。 目的是提供与print(your_model)中的print(your_model)无法提供的信息互补的信息。 用法 pip install torchsummary或 git clone https://github.com/sksq96/pytorch-summary from torchsummary import summary summary ( your_model , in


【文件预览】:
pytorch-summary-master
----LICENSE(1KB)
----setup.py(347B)
----README.md(7KB)
----.gitignore(26B)
----torchsummary()
--------torchsummary.py(5KB)
--------__init__.py(50B)
--------tests()

网友评论