随笔分类 -  Pytorch

摘要:1.pytorch主要的包: torch: 最顶层包及张量库 torch.nn: 子包,包括模型及建立神经网络的可拓展类 torch.autograd: 支持所有微分操作的函数子包 torch.nn.functional: 其他所有函数功能,包括激活函数,卷积操作,构建损失函数等 torch.opt 阅读全文
posted @ 2020-06-08 17:46 Parallax 阅读(1857) 评论(0) 推荐(1)
摘要:torch中Variable和Tensor合并 4 import torch 5 from torch.autograd import Variable # torch 中 Variable 模块 6 7 tensor = torch.FloatTensor([[1,2],[3,4]]) 8 var 阅读全文
posted @ 2020-04-27 15:31 Parallax 阅读(384) 评论(0) 推荐(0)
摘要:Pytorch有很多方便易用的包,今天要谈的是torchvision包,它包括3个子包,分别是: torchvison.datasets ,torchvision.models ,torchvision.transforms ,分别是预定义好的数据集(比如MNIST、CIFAR10等)、预定义好的经 阅读全文
posted @ 2020-03-03 10:08 Parallax 阅读(2580) 评论(0) 推荐(1)
摘要:https://github.com/pytorch/vision/tree/master/torchvision/models 几乎所有的常用预训练模型都在这里面 总结下各种模型的下载地址: 1 Resnet: 2 3 model_urls = { 4 'resnet18': 'https://d 阅读全文
posted @ 2020-02-25 18:22 Parallax 阅读(4604) 评论(0) 推荐(2)