2022年7月7日
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 阅读(1485) 评论(0) 推荐(0) 编辑