博客
关于我
卷积网络的运算量和参数量的计算
阅读量:504 次
发布时间:2019-03-07

本文共 201 字,大约阅读时间需要 1 分钟。

在实验中,经常需要计算网络的参数量和预算量。

参考:

运算量和参数量的计算代码

# Calculate params & FLOPs    from torchsummaryX import summary    dummy_input = torch.zeros(1, 3, 128, 128).cuda()    summary(model.netG, dummy_input)    exit()

 

 

 

转载地址:http://yvajz.baihongyu.com/

你可能感兴趣的文章
MySQL主从篇:死磕主从复制中数据同步原理与优化
查看>>
mysql主从配置
查看>>
MySQL之2003-Can‘t connect to MySQL server on ‘localhost‘(10038)的解决办法
查看>>
MySQL之CRUD
查看>>
MySQL之DML
查看>>
Mysql之IN 和 Exists 用法
查看>>
MYSQL之REPLACE INTO和INSERT … ON DUPLICATE KEY UPDATE用法
查看>>
MySQL之SQL语句优化步骤
查看>>
MYSQL之union和order by分析([Err] 1221 - Incorrect usage of UNION and ORDER BY)
查看>>
Mysql之主从复制
查看>>