当前标签: Pytorch
PyTorch Hub发布获Yann LeCun强推!一行代码调用经典模型
作者 | Team PyTorch 译者 | Monanfei 责编 | 夕颜 出品 | AI科技大本营(ID: rgznai100) 导读:6月11日,Facebook PyTorch 团队推出了全新 API PyTorch Hub,提供模型的基本构建模块,用于提高机器学习研究的模型...
2019-06-11 14:53
阅读(333)
重磅!Facebook更新PyTorch 1.1,打算跨GPU分割神经网络
时隔半年不到,PyTorch 已经从之前的 1.0 升级到 1.1 版本了。刚刚,Facebook 在年度开发者大会 F8 上宣布正式发布 PyTorch 1.1 版本,这是对 PyTorch 1.0 的一次大的功能升级。 作者 | 琥珀 出品 | AI科技大本营(ID:rgznai100) PyTorc...
2019-05-02 19:32
阅读(276)
pytorch:RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 1 a
<>1.pytorch报错 Namespace(batchSize=16, gpu='0,1', lr=0.001, nepoch=210, num_workers=8) Epoch: 0 Epoch:0 [0|173089] loss:3.747539 Epoch:0 [1|173089] loss:3.629...
2018-11-28 16:39
阅读(2019)
人工智能新书推荐——《PyTorch机器学习从入门到实战》
人工智能从入门到实战, 轻松玩转PyTorch框架。 校宝在线出的新书《PyTorch机器学习从入门到实战》已由机械工业出版社出版!各大平台有售。 书籍详情(点击查看):机械工业出版社 <http://www.cmpbook.com/stackroom.php?id=44729> | 亚马逊 <https:...
2018-11-05 10:30
阅读(579)
PyTorch在NLP任务中使用预训练词向量
在使用pytorch或tensorflow等神经网络框架进行nlp任务的处理时,可以通过对应的Embedding层做词向量的处理,更多的时候,使用预训练好的词向量会带来更优的性能。下面分别介绍使用gensim和torchtext两种加载预训练词向量的方法。 <>1.使用gensim加载预训练词向量 对...
2018-11-01 21:49
阅读(1127)
ubuntu 安装 pytorch (ubuntu16.04+pytorch0.4.1+python2.7+cuda8.0)
这里使用pip安装pytorch,我试过cuda安装,和电脑的配置没兼容,没有安装成功,后来发现使用pip安装很简单方便,就是用pip安装 首先进入pytorch官网 <https://pytorch.org/>,往下拉会看到不同的安装选项 根据自己电脑的配置选择不同的选项,例如我选择的stable版本、Linu...
2018-10-23 00:15
阅读(843)
pytorch 获取tensor维度信息
参考 https://stackoverflow.com/questions/46826218/pytorch-how-to-get-the-shape-of-a-tensor-as-a-list-of-int <https://stackoverflow.com/questions/46826218/pytor...
2018-10-18 20:35
阅读(2030)
PyTorch 常用方法总结4:张量维度操作(拼接、维度扩展、压缩、转置、重复……)
涉及的方法有下面几种 torch.cat() torch.Tensor.expand() torch.squeeze() torch.Tensor.repeat() torch.Tensor.narrow() torch.Tensor.view() torch.Tensor.resize_() torch.Ten...
2018-10-16 15:09
阅读(774)