- 博客(1)
- 收藏
- 关注
原创 【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
565
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人