随笔分类 - Pytorch
摘要:1.pytorch主要的包: torch: 最顶层包及张量库 torch.nn: 子包,包括模型及建立神经网络的可拓展类 torch.autograd: 支持所有微分操作的函数子包 torch.nn.functional: 其他所有函数功能,包括激活函数,卷积操作,构建损失函数等 torch.opt
阅读全文
摘要: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
阅读全文
摘要:Pytorch有很多方便易用的包,今天要谈的是torchvision包,它包括3个子包,分别是: torchvison.datasets ,torchvision.models ,torchvision.transforms ,分别是预定义好的数据集(比如MNIST、CIFAR10等)、预定义好的经
阅读全文
摘要:https://github.com/pytorch/vision/tree/master/torchvision/models 几乎所有的常用预训练模型都在这里面 总结下各种模型的下载地址: 1 Resnet: 2 3 model_urls = { 4 'resnet18': 'https://d
阅读全文

浙公网安备 33010602011771号