PyTorch
玄云飘风
这个作者很懒,什么都没留下…
展开
-
PyTorch中的Tensor
生成一个Tensor(t)可以与list或numpy中的array互相转化:t=T.Tensor(list)t=T.Tensor(np.array)t=T.from_numpy(np.array)list=T.tolist(t)array=t.numpy()Tensor和np.array共享内存,所以他们之间的转换很快,而且几乎不会...原创 2018-05-16 16:47:07 · 14451 阅读 · 1 评论 -
统计官方模型的参数量和计算量
统计官方模型参数量和计算量原创 2022-11-22 16:15:02 · 688 阅读 · 1 评论 -
【pytorch】torchvion.transforms.RandomResizedCrop
pytorch文档阅读与分析原创 2022-11-17 14:12:55 · 759 阅读 · 0 评论 -
【PyTorch】保存和载入模型的两种方法
import torchimport argparse# 载入模型和载入参数if torch.cuda.is_available(): if opt.pre_trained_model_type == "model": model = torch.load(opt.pre_trained_model_path) if opt.pre_trained_mode...原创 2019-05-31 18:59:11 · 6602 阅读 · 0 评论 -
【PyTorch1.0】torchvision文档翻译(五)——torchvision.utils
原文:https://pytorch.org/docs/stable/torchvision/utils.htmltorchvision.utils.make_grid(tensor, nrow=8, padding=2, normalize=False, range=None, scale_each=False, pad_value=0)把图片排列成网格形状。参数:tensor(...翻译 2019-02-22 20:07:46 · 868 阅读 · 0 评论 -
【PyTorch1.0】torchvision文档翻译(三)——torchvision.models
原文:https://pytorch.org/docs/stable/torchvision/models.htmlmodels子包定义了以下模型架构:AlexNetVGGResNetSqueezeNetDenseNetInception v3你可以通过调用以下构造函数构造随机权重的模型:import torchvision.models as modelsresnet...翻译 2019-02-22 20:05:56 · 870 阅读 · 0 评论 -
【PyTorch1.0】torchvision文档翻译(二)——torchvision.datasets
原文:https://pytorch.org/docs/stable/torchvision/datasets.html所有的数据集都是torch.utils.data.Dataset的子类, 即:它们实现了__getitem__和__len__方法。因此,它们都可以传递给torch.utils.data.DataLoader,进而通过torch.multiprocessing实现批数据的并...翻译 2019-02-22 20:04:05 · 814 阅读 · 0 评论 -
【PyTorch1.0】torchvision文档翻译(一)——torchvision Reference
原文:https://pytorch.org/docs/stable/torchvision/index.htmltorchvision 包收录了若干重要的公开数据集、网络模型和计算机视觉中的常用图像变换包参考torchvision.datasetsMNISTFashion-MNISTEMNISTCOCOLSUNImageFolderDatasetFolderImag...翻译 2019-02-22 19:58:48 · 587 阅读 · 0 评论 -
【PyTorch1.0】torchvision文档翻译(四)——torchvision.transforms
参考链接https://pytorch.org/docs/stable/torchvision/transforms.htmlfrom torchvision import transformsimport numpy as npfrom PIL import Imagey = Image.new('RGB',(2,2),'red') #随机生成红色PIL图片y...翻译 2018-08-28 16:04:25 · 2029 阅读 · 0 评论 -
PyTorch1.0中的Dataloader
注:以下代码基于PyTorch1.0前面我们说到Dataset,是打包了数据地址、规模和其他的一些非必需功能,让Dataloader来调用。本文我们通过Dataloader的源码来看下原始数据是如何夹在到模型之中的。 def __init__(self, dataset, batch_size=1, shuffle=False, sampler=None, batch_sampler=N...原创 2018-12-25 22:17:41 · 2223 阅读 · 1 评论 -
PyTorch中的Dataset
注:本文源码基于PyTorch1.0,目前在PyTorch0.4下没有发现错误。PyTorch中的Dataset是一个抽象类,我们可以通过继承Dataset来将数据集的源文件、规模和其他非必要的功能打包,从而供DataLoader使用。在这一过程中,须重载的函数包括:__init__():构造函数,略过不说。__getitem__():_DataLoaderIter()类中有调用:# ...原创 2018-12-25 17:06:51 · 7290 阅读 · 1 评论 -
【PyTorch】Tensor和tensor的区别
本文列举的框架源码基于PyTorch1.0,交互语句在0.4.1上测试通过import torch在PyTorch中,Tensor和tensor都能用于生成新的张量:>>> a=torch.Tensor([1,2])>>> atensor([1., 2.])>>> a=torch.tensor([1,2])>>>...原创 2018-12-29 14:50:31 · 30265 阅读 · 8 评论 -
PyTorch中Tensor的查找和筛选
本文源码基于版本1.0,交互界面基于0.4.1import torch按照指定轴上的坐标进行过滤index_select()沿着某tensor的一个轴dim筛选若干个坐标>>> x = torch.randn(3, 4) # 目标矩阵>>> xtensor([[ 0.1427, 0.0231, -0.5414, -1.0009], ...原创 2018-12-29 00:03:13 · 62705 阅读 · 8 评论 -
【pytorch】torch.utils.data.DataLoader中的pin_memory属性
doc解释:https://pytorch.org/docs/stable/_modules/torch/utils/data/dataloader.html#DataLoaderpin_memory (bool, optional): If True, the data loader will copy tensors into CUDA pinned memory before retu...原创 2018-10-22 19:49:00 · 7113 阅读 · 3 评论 -
【PyTorch】常用操作(持续更新)
保存和加载整个模型torch.save(net, 'model.pkl')net = torch.load('model.pkl')仅保存和加载模型参数state = { 'net': net.state_dict(), 'acc': acc, 'epoch': epoch,}torch.save(state, 'params.pkl')torch.save...原创 2018-11-20 21:19:51 · 223 阅读 · 0 评论