
Pytorch
Pytorch
Ztomepic
这个作者很懒,什么都没留下…
展开
-
win10跑Pytorch模型出现RuntimeError: CUDA out of memory的解决方法
Kaggle官方提供的GPU频繁断开,被迫在自己的机器上跑Kernal,显卡是贫民配置1050TI,显存4GB。训练无法开始,直接报错RuntimeError: CUDA out of memory,尝试多种方法后解决,现在能够龟速训练了。1.修改Dataloader中的num_worker参数为0(似乎仅针对windows系统下)2.缩小batch_size3.把训练的代码块放入if __...原创 2019-12-29 15:59:36 · 3124 阅读 · 3 评论 -
Pytorch-1:关于Tensor
TensorTensor-张量,是一种数据类型,基本上来说是多维array的别名,这里的“多维”可以是任何维数,也就是说1d-tensor → vector。Rank,Axes and shapeTensor的三个基本属性:Rank,Axes and shape。**Rank:**维数,其大小恰好说明了需要多少个Index来取得Tensor里的一个数。Axes: axis复数形式,sh...原创 2020-01-16 23:03:07 · 271 阅读 · 0 评论 -
Pytorch-2:张量关于shape的操作
flattenreshapesqueeze explainedreshape原创 2020-01-10 18:04:17 · 3344 阅读 · 0 评论 -
Pytorch-3:element-wise operation(包括element-wise大小比较)
Element-wise operationElement-wise 也称为 Componet-wise 或者 Point-wise这种操作在匹配的元素间起作用,“匹配”的含义是:张量有同样的shape元素的“位置”相同t1 = torch.tensor([ [1, 2], [3, 4] ])t2 = torch.tensor([ [9, 8], [7, 6] ])...原创 2020-01-10 18:38:44 · 4525 阅读 · 0 评论 -
Pytorch-4 :Tensor Reduction Operation
Element-wise Operation是多个 tensor 间的操作,Reduction Operation 是一个tensor内部的操作。Reduction OperationReduction Operation的返回值总是使 tensor 的元素数量变小。t = torch.tensor([[0, 1, 0],[2, 0, 2],[0, 3, 3]], dtype=tor...原创 2020-01-10 19:57:56 · 296 阅读 · 0 评论 -
Pytorch-5 : 视觉深度学习prepare data - ETL流程,Dataset 和 DataLoader类
ETLExtract : 获得图片数据Transfrom :将图片转换为tensor形式Load : 将数据转为对象,以便于使用为了实现以上流程,PyTroch提供这两个类:torch.utils.data.Dataset : 抽象类(abstract class),代表datasettorch.utils.data.DataLoader : 包装dataset,从而方便诸...原创 2020-01-11 19:30:09 · 811 阅读 · 0 评论 -
Pytorch-6:建立一个CNN
继承 nn.Module ,建立一个 network class在 network class 的构造函数中定义== layers 为 类的属性(attributes)==用 layers 属性和 nn.functional 的方法来定义forward()torch.nn.Module 类神经网络 和 神经网络的layers 都继承自 torch.nn.module 。forward...原创 2020-01-16 23:49:37 · 423 阅读 · 0 评论 -
Pytorch-7:The Train Process 和 Comfusion Matrix
已有的代码:import torchimport torch.nn as nnimport torch.nn.functional as Fimport torchvisionimport torchvision.transforms as transformstorch.set_printoptions(linewidth=120)train_set = torchvisio...原创 2020-02-05 15:51:33 · 502 阅读 · 0 评论 -
Pytorch之上采样
nn.UpsampleCLASS torch.nn.Upsample(size=None, scale_factor=None, mode='nearest', align_corners=None)参数:size – 根据不同的输入类型制定的输出大小scale_factor – 指定输出为输入的多少倍数。如果输入为tuple,其也要制定为tuple类型。注意:size跟sca...原创 2020-01-01 19:52:32 · 875 阅读 · 0 评论