pytorch
咖豆丘
这个作者很懒,什么都没留下…
展开
-
pytporch(3) 保存和提取 &&批处理
net = torch.nn.Sequential( torch.nn.Linear(2, 10), torch.nn.ReLu(), torch.nn.Linear(10, 2) ) plt.ion() plt.show() optimizers = torch.optim.SGD(net.parameters(), lr=0.02) loss_func = torch.nn.CrossEntropyLoss() #计算的是概率 for t in range(...原创 2020-09-16 08:57:39 · 190 阅读 · 0 评论 -
pytorch(2) 0/1简单分类
import torch from torch.autograd import Variable import matplotlib.pyplot as plt n_data = torch.ones(100, 2) # print("n_data is %s" % n_data) x0 = torch.normal(2*n_data, 1) y0 = torch.zeros(100) x1 = torch.normal(-2*n_data,1) y1 = torch.ones(100) x = t.原创 2020-09-15 20:38:22 · 521 阅读 · 0 评论 -
pytorch(1) 预测曲线
import torch from torch.autograd import Variable import matplotlib.pyplot as plt x = torch.unsqueeze(torch.linspace(-1, 1, 100), dim=1) y = x.pow(2)+0.2*torch.rand(x.size()) x, y = Variable(x), Variable(y) class Net(torch.nn.Module): def __init__(s.原创 2020-09-15 17:09:22 · 378 阅读 · 1 评论