- 博客(2)
- 收藏
- 关注
原创 网络结构搜索之强化学习
算力和需求的增长使得神经架构搜索( Neural Architecture Search,NAS)重回人们的视线。NAS 可迁移性更强,能够以快速且简单粗暴的方式获得高精度模型,成为工程师手中称手的砖块。强化学习(RL)可谓 NAS 中的核武器,谷歌开启并引领该方向的工作。由初期燃烧的显卡到专用硬件,对效率的追求犹如挖矿一般。
2019-07-30 14:36:26 3547 2
原创 THOP: 统计 PyTorch 模型的 FLOPs 和参数量
THOP 是 PyTorch 非常实用的一个第三方库,可以统计模型的 FLOPs 和参数量。使用方法为:from thop import clever_formatfrom thop import profileclass YourModule(nn.Module): # your definitiondef count_your_model(model, x, y): ......
2019-07-30 10:15:36 37033 11
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人