PyTorch之模型的保存与加载

从一个小demo讲解python import torch import matplotlib.pyplot as plt from torch.autograd import Variable # 生成数据 x = torch.unsqueeze(torch.linspace(-1, 1, 100), dim = 1) y = x.pow(2) + 0.2 * torch.rand(x.si
相关文章
相关标签/搜索