【pytorch】神经网络深度学习模型梯度加载注入?自定义注入模型梯度?(含模型参数、梯度的获取与加载注入
模型梯度注入加载问题复现论文的时候一直在思考模型梯度注入的方式,对于模型参数加载很容易得知:#方式1:state_dict()后注入def model_change1(model,param): """ model:被注入的模型 param:要注入model的参数,此例子为与参数同size """ w_avg = copy.deepcopy(model.state_dict()) for key in w_avg.keys(): w_
原创
2022-05-10 14:58:11 ·
571 阅读 ·
0 评论