pytroch保存和提取網絡模型

1.保存整個模型

torch.save(net, 'net.pkl')

將整個網絡保存下來，直接讀取就可以使用。保存的pkl文件較大

2.讀取整個模型

net = torch.load('net.pkl')

3.只保存網絡的參數等

另一種保存網絡的方法是之保存網絡的參數等信息，pkl文件小，讀取也快。但是讀取的時候需要先搭建網絡。

torch.save(net.state_dict(), 'net_parameters.pkl')