文件名称：Jittor-summary：Jittor中的Keras样式model.summary（），从torchsummary修改而来
文件大小：15KB
文件格式：ZIP
更新时间：2024-03-24 05:30:38
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)