01 2024 档案

判断两个权重文件pth是否相同
摘要：方法1: 先加载两个pth： weights1 = torch.load("A.pth") weights2 = torch.load("A.pth") 查看keys: weights1.keys()，可能有state_dict, optimizer等， 通过字符串格式化，str(weights1[ 阅读全文
posted @ 2024-01-31 16:43 嶙羽 阅读(188) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示