pytorch ,可视化1,输出每一层的名字,输出shape,参数量
from torchsummary import summary summary(net, (3, 256, 256))
上一篇 list索引 [:]和len()基础小例子
下一篇 定义一个生成器函数,用代码写一个和range函数功能相同的函数,re模块中函数的使用
版权所有 (c)2021-2022 MSHXW.COM
ICP备案号:晋ICP备2021003244-6号