分类：numpy矩阵

2021
03-05

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

目的：把训练好的pth模型参数提取出来，然后用其他方式部署到边缘设备。Pytorch给了很方便的读取参数接口：nn.Module.parameters()直接看demo：fromtorchvision.models.alexnetimportalexnetmodel=alexnet(pretrained=True).eval().cuda()parameters=model.parameters()forpinparameters:numpy_para=p.detach().cpu().numpy()print(type(numpy_para))print(numpy_para.shape)上面得到的numpy_para就是numpy参数... 继续阅读 >