Pytorch获取梯度、修改网络参数

pytorch 获取某一参数x计算的梯度, x.grad

pytorch修改网络参数:pytorch不支持直接修改某一参数,但是可以将model 的state_dict取出来,赋值给另一变量model_state_dict,在model_state_dict中进行修改以后再用model.load_state_dict(model_state_dict)加载,就可以完成修改。

尊重原作,转载请注明,转载自:https://blog.csdn.net/kr2563

猜你喜欢

转载自blog.csdn.net/kr2563/article/details/109546065
今日推荐