
Pytorch
阿瓜瓜瓜
这个作者很懒,什么都没留下…
展开
-
torch-sparse安装问题
无论怎么修改环境变量就是搞不成,简单粗暴点,直接在终端输入以下命令,使用临时变量,路径即cuda的安装路径export CUDA_HOME=/usr/local/cuda-10.0torch-sparse、torch_geometric、torch_scatter、torch_cluster等版本问题参考的这位老哥的博客...原创 2020-07-14 15:06:34 · 892 阅读 · 0 评论 -
cuda() missing 1 required positional argument: 'self'
新开一个分类记录一下自己调代码的时候遇到过的bugcuda() missing 1 required positional argument: 'self'今天遇到这样一个bug,模型在转到GPU上时,说缺少self参数??一脸懵逼,类已经实例化了怎么可能没有self,仔细一看自己的实例化代码,果然不对,缺了个括号。。。。。。。。。。。。。。。这不智障嘛,不带括号相当于只是把网络放到了G...原创 2019-10-30 15:32:14 · 1047 阅读 · 0 评论 -
torch.utils.data.Dataset类和torch.utils.data.DataLoader类
torch.utils.data.Dataset类和torch.utils.data.DataLoader类转载 2019-10-20 09:31:01 · 480 阅读 · 0 评论 -
Pytorch框架猫狗大战之模型定义与调用
模型调用首先再models/__init__.py中,有以下代码:from .alexnet import AlexNetfrom .resnet34 import ResNet34from .squeezenet import SqueezeNet# from torchvision.models import InceptinV3# from torchvision.mode...原创 2019-10-18 11:51:46 · 552 阅读 · 0 评论 -
Pytorch框架猫狗大战之数据预处理
数据预处理阶段,dataset.py中主要定义了DogCat类,并定义了__init__、__getitem__、__len__三个类方法,话不多说直接上代码,具体解读都在每一行后面的注释中。# coding:utf8import osfrom PIL import Imagefrom torch.utils import dataimport numpy as npfrom to...原创 2019-10-18 10:58:25 · 495 阅读 · 0 评论 -
Pytorch框架猫狗大战之先跑起来!
最近在跟着陈云老师的书《深度学习框架Pytorch:入门与实践》学习Pytorch,之前也有学过在B站学过吴恩达老师的machine learning课程,到现在还没有看完,有的地方看不懂看不下去就跳去莫烦小哥的视频学习,跳来跳去感觉自己学到了一点皮毛但仍旧不会用,之后又在室友在推荐下看了《Neural Networks and Deep Learning》入门深度学习,看到最后一章的时候代码用到...原创 2019-10-17 22:50:30 · 972 阅读 · 1 评论 -
sort VS sorted
Python之sort VS sorted转载 2019-10-17 19:34:45 · 170 阅读 · 0 评论 -
os.listdir() 方法和os.path.join() 函数
os.listdir() 方法和os.path.join() 函数转载 2019-10-17 19:23:12 · 483 阅读 · 0 评论 -
torchvision库简介
Pytorch之torchvision库简介转载 2019-10-17 11:49:11 · 937 阅读 · 0 评论