PyTorch中的模型摘要类似于Keras中的`model.summary()`-Python开发

时间:2021-05-25 12:29:21
【文件属性】:
文件名称:PyTorch中的模型摘要类似于Keras中的`model.summary()`-Python开发
文件大小:7KB
文件格式:ZIP
更新时间:2021-05-25 12:29:21
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)

网友评论