pytorch
文章平均质量分 58
沉思的雅典娜
我是一个爱学习的人
展开
-
import torchvision jupyter或者jupyter-lab崩溃(crash)
问题描述:当我在jupyter-lab中import torchvision的时候,jupyter kernel died,网上查了半天资料发现torch和torchvision安装有问题,我想很多人也可能会碰到这种问题,因此,写下这篇内容。原因:可能是需要安装cpu版本的包解决办法:第一步:中国大陆的朋友之下下面代码会加快安装速度。conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud原创 2022-05-09 16:26:23 · 785 阅读 · 0 评论 -
pytorch生成带滑动的训练集测试集(batch)
话不多说,先上代码Example1from torch.utils import dataimport torchbatch_size = 3t1 = np.arange(0,20)t2 = np.arange(0,20)+100生成序列0~19和100~190,分别模拟x_train和y_traintensor1 = torch.from_numpy(t1)tensor2 = torch.from_numpy(t2)上一步很关键,否则报错data_set = da原创 2022-04-14 17:16:46 · 1547 阅读 · 0 评论 -
PyTorch backward原理
第一部分x = torch.tensor([2., 1.], requires_grad=True).view(1,2)print(x)y = torch.tensor([[1., 2.], [3., 4.]], requires_grad=True)print(y.shape)z = torch.mm(x, y)print(f"z:{z}")print('*****************')grad_tensor = torch.Tensor([[1,0]])print(z.原创 2022-04-08 18:22:43 · 7668 阅读 · 1 评论 -
tensor view(-1,xx)使用-1什么缘由
参数-1是什么意思?如果你不知道你想要多少行,但确定列数,那么你可以将行数设置为-1(你可以将它扩展到具有更多维度的张量。只有一个轴值可以是-1)。这是告诉系统Library:给我一个具有这么多列的张量,并计算实现这一点所需的适当行数参考https://vimsky.com/article/3888.html...原创 2019-11-15 20:37:36 · 1806 阅读 · 0 评论