莫烦Pytorch之保存加载网络

import torch from torch.autograd import Variable import matplotlib.pyplot as plt x=torch.unsqueeze(torch.linspace(-1,1,1000),dim=1) y=x.pow(2)+0.2*torch.rand(x.size()) def save(): net1=torch.nn.
相关文章
相关标签/搜索