判断两个权重文件pth是否相同

方法1: 先加载两个pth:

1
2
weights1 = torch.load("A.pth")
weights2 = torch.load("A.pth")

 查看keys: weights1.keys(),可能有state_dict, optimizer等,

通过字符串格式化,str(weights1['state_dict'])==str(weights2['state_dict']),即可逐项比较;

方法2: 比较md5sum A.pth和md5sum B.pth,返回的序列是否一致

 

 

参考:https://blog.csdn.net/weixin_40519680/article/details/129533896

posted @   嶙羽  阅读(188)  评论(0编辑  收藏  举报
编辑推荐:
· AI与.NET技术实操系列:基于图像分类模型对图像进行分类
· go语言实现终端里的倒计时
· 如何编写易于单元测试的代码
· 10年+ .NET Coder 心语,封装的思维:从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热?
阅读排行:
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 25岁的心里话
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现
点击右上角即可分享
微信分享提示