1. 首页
  2. 编程语言
  3. Python
  4. 从Pytorch模型pth文件中读取参数成numpy矩阵的操作

从Pytorch模型pth文件中读取参数成numpy矩阵的操作

上传者: 2022-01-13 00:53:51上传 PDF文件 52.14 KB 热度 6次

Pytorch给了很方便的读取参数接口:直接看demo:上面得到的numpy_para就是numpy参数了~model.parameters()是以一个生成器的形式迭代返回每一层的参数。所以用for循环读取到各层的参数,循环次数就表示层数。而每一层的参数都是方便又好用,爆赞~将python训练好的.pth文件转为.pt

用户评论