莫煩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.
相關文章
相關標籤/搜索