pytorch
先生的小花
这个作者很懒,什么都没留下…
展开
-
关于tensor类型数据却报错‘list‘ object has no attribute ‘to‘
先放图:问题所在:我的数据本事应该是tensor类型,可以to(device),但是却报错是list(列表)所以无法’to’。于是在出错前的语句先print(images),发现是[tensor[…], …],那问题很简单,使用images = torch.stack(images, dim=1),将列表元素取出(即list类型转换成tensor类型),大功告成!...原创 2021-06-29 17:27:03 · 14236 阅读 · 2 评论 -
服务器创建虚拟环境并安装pytorch
上一次提到Linux服务器通过pipenv安装虚拟环境。**但当服务器本身装了anaconda,而你只是想建一个属于自己的envs时,其实conda是支持的,**所以!根本!不建议用pipenv!(个人意见)(如果大家两个都装过,就知道我为什么这么说了~)创建新的虚拟环境命令$ conda create -n "env name" python=版本号激活虚拟环境命令$ source activate "env name"安装pytorch(这里我还想装torcvision等,所以原创 2021-06-04 16:43:41 · 801 阅读 · 0 评论 -
pytorch_resnet34模型调用及参数修改
pytorch是包含一些常见的神经网络模型的,ResNet34、ResNet18、VGG等等,都在models模块中,调用接口如下:model = models.resnet34(pretrained=True, progress=True)预训练pretrained和进度条progress设为True或False就看你自己需求咯~⚠️与此同时,你可能还会遇到pytorch官网的输出类别数与你所需不同:你想分七类,可resnet34官网是1000类,那就调整一下呗:# 修改最后线性层的输出通道数原创 2021-04-12 21:48:53 · 4657 阅读 · 0 评论