会员
周边
众包
新闻
博问
闪存
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
Masked Prometheus
machine learning, algorithm, social computing...
博客园
首页
新随笔
联系
订阅
管理
2021年6月19日
pytorch中的add_module函数
摘要: 现只讲在自定义网络中add_module的作用。 总结: 在自定义网络的时候,由于自定义变量不是Module类型(例如,我们用List封装了几个网络),所以pytorch不会自动注册网络模块。add_module函数用来为网络添加模块的,所以我们可以使用这个函数手动添加自定义的网络模块。当然,这种情
阅读全文
posted @ 2021-06-19 17:32 蒙面的普罗米修斯
阅读(10429)
评论(0)
推荐(3)
编辑
公告