文章目录1.前言2.torch.save(保存模型)3.torch.load整个网络4.torch.load网络参数(只提取参数)5.调用三个函数
1.前言
训练好了一个模型, 我们当然想要保存它, 留到下次要用的时候直接提取直接用,下面我将来讲如何存储训练好的模型参数
2.torch.save(保存模型)
首先,先搭建一个神经网络
import torch
from torch import nn
import matplotlib.pyplot as plt
torch.manual_see