PyTorch
-
PyTorch余弦学习率衰减
今天用到了PyTorch里的CosineAnnealingLR,也就是用余弦函数进行学习率的衰减。 下面讲讲定义CosineAnnealingLR这个类的对象时输入的几个参数是什么,代码示例就不放了。 正文 torch.optim.lr_scheduler.CosineAnnealingLR(optimizer, T_max, eta_min=0, last…
-
windows10 安装 Anaconda 并配置 pytorch1.0
官网下载Anaconda安装包,按步骤安装即可安装完后,打开DOS,或Anaconda自带的Anaconda Prompt终端查看Anaconda已安装的安装包C:\Users\jiangshan>conda list安装 matplotlibC:\Users\jiangshan>conda install matplotlib设置镜像# 添加A…
-
CentOS7.5服务器安装(并添加用户) anaconda3 并配置 PyTorch1.0
===========================================================================================[admin@localhost ~]$ sudo vim /etc/sudoersWe trust you have received the usual lecture fr…
-
VScode中pytorch出现Module ‘torch’ has no ‘xx’ member错误
因为代码变量太多,使用Sublime text并能很好地跳转,所以使用VsCode 神器。 导入Pytorch模块后出现了 Module ‘torch’ has no cat member,所以在网上找解决办法,这位博主的文章很好用,一路解决。 我的版本python3.7无Anacada,解决办法,打开设置,搜索pyt…
-
Pytorch 扩展Tensor维度、压缩Tensor维度
相信刚接触Pytorch的宝宝们,会遇到这样一个问题,输入的数据维度和实验需要维度不一致,输入的可能是2维数据或3维数据,实验需要用到3维或4维数据,那么我们需要扩展这个维度。其实特别简单,只要对数据加一个扩展维度方法就可以了。 1.1 torch.unsqueeze(self: Tensor, dim: _int) torch.unsqueez…
-
windows下配置pytorch环境
借鉴了B站大佬的视频,自己总结安装如下。 首先安装anaconda 按照操作顺序,依次安装,按照我个人习惯,不喜欢讲文件安装在C盘,你们自己决定。 安装完毕之后。 之后打开Anaconda Prompt,如下图: 换源: 输入红线命令到Anaconda Prompt中,不过我建议换清华源,因为用conda安装软件时,比如安装pytroch就会很容易中断,因为…
-
OpenCV加载Pytorch模型出现Unsupported Lua type 解决方法
原因 Torch有两个版本,一个就叫Torch一个专门给Python用的Pytorch,它们训练完之后保存下来的模型是不一样的.说到这问题就很清楚了.OpenCV的ReadNetFromTorch支持的是前者… 解决方法 那么有没有解决办法呢,答案是有的.PyTorch支持把模型保存为ONNX格式.而这个格式在opencv是支持的.操作如下: impor…
-
[Pytorch]Pytorch中图像的基本操作(TenCrop)
转自:https://www.jianshu.com/p/73686691cf13 下面是几种常写的方式 第一种方式 normalize = transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225]) transformList = [] transformList.append(tr…
-
[pytorch]pytorch loss function 总结
原文: http://www.voidcn.com/article/p-rtzqgqkz-bpg.html 最近看了下 PyTorch 的损失函数文档,整理了下自己的理解,重新格式化了公式如下,以便以后查阅。 注意下面的损失函数都是在单个样本上计算的,粗体表示向量,否则是标量。向量的维度用 表示。 nn.L1Loss nn.SmoothL1Loss 也叫作 …
-
pytorch转onnx问题
Fail to export the model in PyTorch https://github.com/onnx/tutorials/blob/master/tutorials/PytorchAddExportSupport.md#fail-to-export-the-model-in-pytorch 1. RuntimeError: ONNX ex…