pytorch修改模型参数的方法

pytorch貌似不支持直接修改网络参数

参考了知乎https://zhuanlan.zhihu.com/p/158876093和评论,自己测试了一下,有用的方法如下:
(我这里是修改模型参数的数据类型)

for p in model.parameters():
    p.data=p.data.to(torch.float16)

记录一下

你可能感兴趣的:(pytorch,pytorch,深度学习)