![](https://img-blog.csdnimg.cn/20201014180756780.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
人工智能
关于机器学习、深度学习、图神经网络的一些基本知识与最新进展
Javy Wang
学无止境 气有浩然
展开
-
PyTorch中的设备类型问题 device type cuda
用pytorch包搭建神经网络时,经常遇到GPU和CPU的指定问题,有时候明明指定了GPU(cuda)却依然会报错,下面将这些常见的错误整理出来。问题一:目标参数的设备类型本应该是cuda而实际上却是CPURuntimeError: Expected object of device type cuda but got device type cpu for argument #2 ‘target’除了上面博文提到的三种情况以外,还有一种可能是变量的设备指定方式有问题,如下:device = tor原创 2020-10-29 14:03:33 · 7005 阅读 · 1 评论 -
图卷积网络详细介绍(一)
在图上进行机器学习是一项困难的任务,因为它的复杂性很高,同时也是由于信息丰富的图结构。这篇文章是关于如何用图卷积网络(GCNs)对图进行深度学习的系列文章中的第一篇。本文翻译自Tobias Skovgaard Jepsen写在Medium上的介绍文章,已经获得其本人的许可。翻译 2018-11-09 19:17:24 · 18085 阅读 · 25 评论