博客
关于我
卷积网络的运算量和参数量的计算
阅读量: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新建用户和数据库并授权
查看>>
mysql日志
查看>>
mysql日志 事务问题_mysql因为事务日志问题无法启动
查看>>
mysql日志文件
查看>>
mysql日志管理学习笔记
查看>>
mysql日志问题定位实用命令
查看>>
MySQL日期时间函数大全
查看>>
mysql时间相减的问题
查看>>