会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
tangjunjun
博客园
首页
新随笔
联系
订阅
管理
2022年3月15日
权重pth读取更换其值重新保存方法(附带参数计算)
摘要: 本文主要解决模型权重迁移,主要使用pytorch读取某个权重,将其赋值给新权重格式,以下为原始代码: 顺带参数计算函数代码: 参数计算: def count_parameters(model): return sum(p.numel() for p in model.parameters() if
阅读全文
posted @ 2022-03-15 16:33 tangjunjun
阅读(929)
评论(0)
推荐(0)
编辑
公告
https://rpc.cnblogs.com/metaweblog/tangjunjun