判断两个权重文件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  嶙羽  阅读(62)  评论(0)  编辑  收藏  举报