会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
嶙羽
博客园
首页
新随笔
联系
订阅
管理
2021年3月12日
model.add_module
摘要: Pytorch使用这个方法,就可以在已有模型的基础上重新加一些结构,比如已经加载了某个预训练模型,但仍需要增加一些模块,此时已经不能在一开始构建模型的时候init进去模块,于是可以在load完模型后,用model.add_module,把所需要的模块增添进去,使用方法是model.add_modul
阅读全文
posted @ 2021-03-12 09:41 嶙羽
阅读(1096)
评论(0)
推荐(0)
编辑
公告