2024年1月31日
判断两个权重文件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 嶙羽 阅读(163) 评论(0) 推荐(0) 编辑