会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
tangjunjun
博客园
首页
新随笔
联系
订阅
管理
2020年9月25日
基于pytorch框架对神经网络权重初始化(inite_weight)方法详解
摘要: 今天重新研究了一下pytorch如何自定义权重,可以根据条件筛选赋值,也可以根据自定义某个张量赋值,还可以根据pth文件部分赋值等情况, 我总结了三种方法(self.modules(),self.state_dict(),self.named_parameters()),每种方法有约2种情况,希望对大家有所帮助, 然具体各自函数代表什么,可自行查阅,如self.parameters函数,我给出具体实现权重的初始化方法(已使用代码验证)。
阅读全文
posted @ 2020-09-25 17:05 tangjunjun
阅读(10332)
评论(0)
推荐(1)
编辑
公告
https://rpc.cnblogs.com/metaweblog/tangjunjun