pytorch打印模型每层的结构
可以用torchsummary下的summary方法，或者torchinfo下的summary方法，两者功能是一样的以torchsummary下的summary方法为例说明首先通过pip install torchsummary之后在代码中：from torchsummary import summarysummary(model, (1, 28, 28)) #分别是输入数据的三个维度...
·
可以用torchsummary下的summary方法，或者torchinfo下的summary方法，两者功能是一样的
以torchsummary下的summary方法为例说明
首先通过pip install torchsummary
之后在代码中：
from torchsummary import summary
summary(model, (1, 28, 28)) #分别是输入数据的三个维度
技术共进，成长同行——讯飞AI开发者社区
更多推荐
2
0- 0
已为社区贡献8条内容
所有评论(0)