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

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