pytorch统计模型参数量
摘要:用resnet50 来举例子 其中numel表示含有多少element,通过此操作可以统计模型的参数量有多少 另外,两个是一样的,方便debug看其中到底有什么东西
阅读全文
posted @ 2018-08-24 14:36
写博客的一个目的是记录和成长
posted @ 2018-08-24 14:36
posted @ 2018-08-17 22:26
posted @ 2018-08-14 10:44
posted @ 2018-08-07 16:51
posted @ 2018-08-07 16:28