pytorch
从天而降小可爱
这个作者很懒,什么都没留下…
展开
-
Pytorch使用cuda报错RuntimeError:cuda runtime error(999)
在终端重新输入sudo rmmod nvidia_uvm # 删除nvidia模块sudo modprobe nvidia_uvm # 重新加载模块原创 2022-05-10 15:33:23 · 661 阅读 · 0 评论 -
pytorch使用新版本执行旧代码出现错误
问题:module.norm1.norm_func.running_mean” and “module.norm1.norm_func.running_var” for InstanceNorm2d with track_running_stats=False. If state_dict is a checkpoint saved before 0.4.0, this may be expected because InstanceNorm2d does not track running stat原创 2020-11-05 21:13:27 · 390 阅读 · 0 评论 -
torchvision.utils的解析
torchvision.utilssave_imagemake_gridtorchvision.utils.make_grid(tensor, nrow=8, padding=2, normalize=False, range=None, scale_each=False, pad_value=0)把图片排列成网格形状。参数:tensor(Tensor 或 list)– 四维批(ba...原创 2020-04-13 11:17:56 · 2774 阅读 · 0 评论 -
pytorch中ConvTranspose2d的计算公式
参考文献:pytorch中ConvTranspose2d的计算公式class torch.nn.ConvTranspose2d(in_channels, out_channels, kernel_size, stride=1, padding=0, output_padding=0, groups=1, bias=True, dilation=1)参数的含义如下:in_channels(...原创 2020-03-25 10:24:30 · 974 阅读 · 0 评论 -
pytorch中的gather函数
b = torch.Tensor([[1,2,3],[4,5,6]])print bindex_1 = torch.LongTensor([[0,1],[2,0]])index_2 = torch.LongTensor([[0,1,1],[0,0,0]])print torch.gather(b, dim=1, index=index_1)print torch.gather(b, di...原创 2019-12-19 10:22:54 · 170 阅读 · 0 评论 -
python,pytorch的零碎函数知识点
1,isinstance描述:isinstance() 函数来判断一个对象是否是一个已知的类型,类似 type()。区别:isinstance() 与 type() 区别:type() 不会认为子类是一种父类类型,不考虑继承关系。isinstance() 会认为子类是一种父类类型,考虑继承关系。如果要判断两个类型是否相同推荐使用 isinstance()。语法:isinst...转载 2019-02-19 09:39:54 · 379 阅读 · 0 评论 -
pytorch中的pre-train函数模型引用及修改
转载:https://blog.csdn.net/whut_ldz/article/details/78874977?utm_source=copy 一、pytorch中的pre-train模型卷积神经网络的训练是耗时的,很多场合不可能每次都从随机初始化参数开始训练网络。pytorch中自带几种常用的深度学习网络预训练模型,如VGG、R...转载 2018-09-22 18:44:50 · 1348 阅读 · 0 评论 -
使用Pytorch实现ResNet
转载:https://blog.csdn.net/kongshuchen/article/details/72285709 ResNet要解决的问题 深度学习网络的深度对最后的分类和识别的效果有着很大的影响,所以正常想法就是能把网络设计的越深越好,但是事实上却不是这样,常规的网络的堆叠(plain network)在网络很深的时候,效果却越来越差了。 其中的原因之一即是网络越深,梯度消失的现...转载 2018-09-14 15:22:54 · 4957 阅读 · 0 评论 -
Pytorch解读ResNet源码
转载:https://blog.csdn.net/u014380165/article/details/79119664 PyTorch框架中有一个非常重要且好用的包:torchvision,该包主要由3个子包组成,分别是:...转载 2018-09-13 14:55:45 · 6451 阅读 · 1 评论 -
resnet网络&解决输入图像大小问题
转载:https://blog.csdn.net/jiangpeng59/article/details/79609392首先看张核心的resnet层次结构图(图1),它诠释了resnet18-152是如何搭建的,其中resnet18和resnet34结构类似,而resnet50-resnet152结构类似。下面先看resnet18的源码 图1resnet18 首先是models....转载 2018-09-12 20:24:43 · 61204 阅读 · 8 评论 -
Pytorch中函数torchvision总结
1. transform函数 torchvision.transforms是pytorch中的图像预处理包 一般用Compose把多个步骤整合到一起: transform = transforms.Compose([ transforms.Resize(256), transforms.RandomCrop(224), transfor...原创 2018-09-16 09:21:32 · 3081 阅读 · 1 评论 -
pytorch中的pretrain模型网络结构修改
(继)pytorch中的pretrain模型网络结构修改继上篇文章提出的两种预训练模型的修改方法外,今天在这里推荐我新学习到的一种方法:这里还是以resnet模型为参考,在去掉预训练resnet模型的后两层(fc层和pooling层)后,新添加一个反卷积层、池化层和分类层。那么对于网络的修改怎样可以快速的去除model本身的网络层呢?一个继承nn.module的model它包含一...转载 2018-08-15 10:00:38 · 1897 阅读 · 3 评论 -
(转载)pytorch中的pre-train模型
一、pytorch中的pre-train模型卷积神经网络的训练是耗时的,很多场合不可能每次都从随机初始化参数开始训练网络。pytorch中自带几种常用的深度学习网络预训练模型,如VGG、ResNet等。往往为了加快学习的进度,在训练的初期我们直接加载pre-train模型中预先训练好的参数,model的加载如下所示:import torchvision.models as models...转载 2018-08-15 09:58:15 · 2349 阅读 · 0 评论