人工智能
-
pytorch使用过程中遇到的一些问题
问题一 ImportError: No module named torchvision torchvison:图片、视频数据和深度学习模型 解决方案 安装torchvision,参照官网 问题二 安装torchvision过程中遇到 Could not find a version that satisfies the requirement olefil…
-
pytorch将cpu训练好的模型参数load到gpu上,或者gpu->cpu上
假设我们只保存了模型的参数(model.state_dict())到文件名为modelparameters.pth, model = Net() 1. cpu -> cpu或者gpu -> gpu: checkpoint = torch.load(‘modelparameters.pth’) model.load_state_dict(check…
-
利用pytorch复现spatial pyramid pooling层
sppnet不讲了,懒得写。。。直接上代码 1 from math import floor, ceil 2 import torch 3 import torch.nn as nn 4 import torch.nn.functional as F 5 6 class SpatialPyramidPooling2d(nn.Module): 7 r”””ap…
-
pytorch遇到的问题:RuntimeError: randperm is only implemented for CPU
由此,我们找到sample.py,第51行如下图修改
-
Pytorch离线安装方法
由于一些内网环境无法使用pip命令安装python三方库,寻求一种能够离线安装pytorch的方法。 方法 由于是内网,首选使用Anaconda代替Python,这样无需手动配置numpy等额外依赖。 访问pytorch离线下载网址根据系统和CUDA版本选择自己需要的whl文件 一共有两个,pytorch和torchvision,例如win10x64下cud…
-
Colab下pytorch基础练习
Colab Colaboratory 是一个 Google 研究项目,旨在帮助传播机器学习培训和研究成果。它是一个 Jupyter 笔记本环境,并且完全在云端运行,已经默认安装好 pytorch,不需要进行任何设置就可以使用,并且完全在云端运行。详细使用方法可以参考 Rogan 的博客:https://www.cnblogs.com/lfri/p/10…
-
PyTorch入门基础
什么是 PyTorch ? PyTorch是一个python库,它主要提供了两个高级功能: GPU加速的张量计算 构建在反向自动求导系统上的深度神经网络 1. 定义数据 一般定义数据使用torch.Tensor ,可以定义多种类型, tensor的意思是张量,是数字各种形式的总称 import torch # 可以是一个数 x = torch.tensor(…
-
利用BERT得到句子的表示向量(pytorch)
在文本分类和文本相似度匹配中,经常用预训练语言模型BERT来得到句子的表示向量,下面给出了pytorch环境下的操作的方法: 这里使用huggingface的transformers中BERT, 需要先安装该依赖包(pip install transformers) 具体实现如下: import torch from tqdm import tqdm i…
-
Pytorch入门实例:mnist分类训练
#!/usr/bin/env python # -*- coding: utf-8 -*- __author__ = ‘denny’ __time__ = ‘2017-9-9 9:03’ import torch import torchvision from torch.autograd import Variable import torch.utils…
-
7月3日云栖精选夜读:强大的PyTorch:10分钟让你了解深度学习领域新流行的框架
摘要: 今年一月份开源的PyTorch,因为它强大的功能,它现在已经成为深度学习领域新流行框架,它的强大源于它内部有很多内置的库。本文就着重介绍了其中几种有特色的库,它们能够帮你在深度学习领域更上一层楼。 热点热议 惊心动魄!程序员们说这些时刻再也不想经历了 作者:程序猿和媛 Java 的最 今年一月份开源的PyTorch,因为它强大的功能,它现在已经成为深…