首先注意pytorch中模型保存有两种格式,pth和pkl,其中,pth是pytorch默认格式,pkl还支持pickle库,不过一般如果没有特殊需求的时候,推荐使用默认pth格式保存 pytorch中有两种数据保存方法,一种是存储整个模型,一种只存储参数 方法一:存储整个模型 #保存 torch.save(model1, 'net.pth') #读取 model1 = torch.load('net.pth') 方法二:存储模型参数 #保存 torch.save(model.state_dic…