Pytorch修改权重pth文件内容

修改参数值 方法1 dict的类型是collecitons.OrderedDict，是一个有序字典，直接将新参数名称和初始值作为键值对插入，然后保存即可。 #修改前 dict = torch.load('./ckpt_dir//model_0.pth') net.load_state_dict(di
posted @ 2022-07-07 10:36  蜕变C  阅读(1396)  评论(0)  编辑  收藏  举报