我在模型的保存加载---直接序列化时候会遇到的问题

le.PicklingError: Can't pickle <class '__main__.Model'>: attribute lookup Model on __main__ fai

这是关于pickle的报错：

处理方法：

一般情况主要是pytorch版本问题。Model模型搭建时候所遇到的问题

我的解决处理方法是：

构建一个model.py文件

class Model(nn.Module):
    def __init__(self, input_size, output_size):
        super(Model,self).__init__()#调用父类方法
        #穿件实列化两个线性层
        self.liner1=nn.Linear(input_size,input_size*2)
        self.liner2=nn.Linear(input_size*2,output_size)

    def forward(self,inputs):
        inputs=self.liner1(inputs)
        output=self.liner2(inputs)
        return output
    #简单的网络搭建完

在另一个python脚本文件中，再通过from..import的方式把网络导入原文件中：

from model import Model

然后正常操作就可以了。