【文件属性】：
文件名称：Jittor-summary：Jittor中的Keras样式model.summary（），从torchsummary修改而来
文件大小：15KB
文件格式：ZIP
更新时间：2021-03-04 11:43:58
Python
model.summary()中的Keras样式model.summary()
1.用法
git clone https://github.com/liuruiyang98/Jittor-summary.git
from jittorsummary import summary
summary ( your_model , input_size = ( channels , H , W ))
请注意， input_size是通过网络进行前向传递所必需的。
请注意，不支持带有cuda的jittorsummary。
jt.flags.use_cuda = 0
2.例子
2.1 MNIST的CNN
import jittor as jt
from jittor import init
from jittor import nn
from jittorsummary import sum
【文件预览】：
Jittor-summary-main
----LICENSE(1KB)
----.gitignore(2KB)
----README.md(4KB)
----jittorsummary()
--------tests()
--------jittorsummary_test.py(2KB)
--------__init__.py(50B)
--------jittorsummary.py(5KB)