![](https://img-blog.csdnimg.cn/20201014180756918.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
个人学习
*持续学习*
持续学习软件测试ing
展开
-
pytorch回归网络
import torchimport torch.optimimport matplotlib.pyplot as pltimport torch.nn.functional as ffrom torch.autograd import Variable# unsequeeze(param) 增加param的维度,在torch中处理2维的张量x = torch.unsqueeze(torch.linspace(-1, 1, 100), dim=1)# 目标函数表达式y = x.pow(2)原创 2021-06-09 09:51:54 · 408 阅读 · 0 评论 -
pytorch批量训练
新版直接传入参数Data.TensorDataset(x, y)没有GPU的num_workers=2会报错,默认使用0。num_workers:用于数据加载的子进程数目,默认0import torchimport torch.utils.data as Data# 批训练数据的大小BATCH_SIZE = 5x = torch.linspace(1, 10, 10)y = torch.linspace(10, 1, 10)# 新版中TensorDataset直接传入参数即可# .原创 2021-06-09 09:22:44 · 312 阅读 · 0 评论