1.保存整个模型 torch.save(net, 'net.pkl') 将整个网络保存下来,直接读取就可以使用。保存的pkl文件较大 2.读取整个模型 net = torch.load('net.pkl') 3.只保存网络的参数等 另一种保存网络的方法是之保存网络的参数等信息,pkl文件小,读取也快。但是读取的时候需要先搭建网络。 torch.save(net.state_dict(), 'net_parameters.pkl')