pytorch
关注
分享
扫一扫
文章平均质量分 69
加油上学人
算法工程师。擅长算法优化，代码重构，微积分。加油打工人
展开
-
torch模型转onnx
将pytorch训练的模型转为onnx格式原创 2023-08-16 17:27:53 · 345 阅读 · 3 评论 -
cuda 11情况下如何配置pytorch 10.2
由于目前pytorch只能支持到10.2的版本，但ubuntu最新的系统驱动直接支持了cuda 11.0， 并且cuda tooklit支持的默认下载也是11.0。1. 需要先降低cuda tooklit的版本cuda-uninstaller in /usr/local/cuda-11.1/binsudo rm -rf /usr/local/cuda-11.1来自：Nidia或cd /usr/local/cuda-11.0/bin/sudo ./cuda-uninstallersudo原创 2020-10-15 10:56:09 · 9412 阅读 · 1 评论 -
GoogleNews-vectors-negative300.bin.gz
cnn for text classifirar中需要用到一个google的包，使用如下命令下载即可：brew install wgetwget -c "https://s3.amazonaws.com/dl4j-distribution/GoogleNews-vectors-negative300.bin.gz"文件大小在1.6G左右转载 2020-10-14 19:12:52 · 2428 阅读 · 0 评论 -
Ubuntu18.04下安装pytorch详细步骤
说明：使用系统为Ubuntu18.04，CUDA版本为10.2，CUDNN为7.6.51、Anaconda的下载安装到官网进行下载，我下载的是Anaconda3-2020.02-Linux-x86_64.sh之后使用命令 bash Anaconda3-2020.02-Linux-x86_64.sh,一直按回车，知道出现yes位置，之后会默认安装。安装成功后我没有截图，这个没有其他疑问，可以一路执行到底。2、查看自己的Python版本使用anaconda后，系统的默认Python版本会被替换，在终原创 2020-07-22 12:00:52 · 14828 阅读 · 17 评论 -
conda 安装pytorch cpu版
conda GPU的命令如图所示：conda install pytorch torchvision -c pytorchconda CPU的命令如图所示：conda install pytorch-cpu -c pytorch pip3 install torchvision转载 2020-03-18 21:57:36 · 2210 阅读 · 0 评论 -
pytorch之模型迁移
参考的书目为《深度学习之pytorch实战计算机视觉》将猫狗大战中写好的VGGNet模型替换为VGG16，ResNet模型，前面首先给出完整的VGGNet模型代码，之后分别给出最另外两个模型的修改位置，并进行标注说明：import torchimport torchvisionfrom torchvision import datasets, transformsimport torch...原创 2020-01-14 16:17:27 · 649 阅读 · 2 评论 -
torch神经网络模型的搭建
包含：卷积层、激活函数、池化层、全连接层class Model(torch.nn.Module): def __init__(self): super(Model, self).__init__() self.convl = torch.nn.Sequential( torch.nn.Conv2d(1, 64, kernel_siz...原创 2020-01-13 22:15:19 · 1016 阅读 · 0 评论 -
pytorch实现猫狗大战报错Found 0 files in subfolders of: G:/pytorch/DogsVSCats\train的解决办法
实现猫狗大战路径载入的时候报了错误，Found 0 files in subfolders of: G:/pytorch/DogsVSCats\train，意思是没有找到下一级目录。参考的书目为《深度学习之pytorch实战计算机视觉》，书中给出的原话是“新建一个名为DogVSCats的文件夹，在该文件夹下面新建一个名为train和valid的子文件夹，再在子文件下面分别新建一个cat和dog的...原创 2020-01-13 21:36:52 · 1696 阅读 · 3 评论 -
pytorch_torch.transforms常用参数
mnist是灰度图，给出的应该是transforms.Normalize((0.1307 ), (0.3081 ))cifar10是transforms.Normalize((0.4914, 0.4822, 0.4465), (0.2023, 0.1994, 0.2010))image net是([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])...原创 2020-01-13 18:44:51 · 1490 阅读 · 0 评论 -
pytorch-error：output with shape [1, 28, 28] doesn't match the broadcast shape [3, 28, 28]
使用深度学习之pytorch实战计算机视觉进行学习时，在6.4实战手写数字识别时遇到了该问题，报错情况如下：Traceback (most recent call last): File "G:/pytorch/torch_数字识别器.py", line 19, in <module> images, lables = next(iter(data_loder_train...原创 2020-01-12 23:19:43 · 2421 阅读 · 1 评论 -
torch.nn中常用的损失函数及使用方法
1）**torch.nn.MSELoss()：**使用均方差函数对损失值进行计算定义类时不需要传入参数，调用的时候需要传入参数import torchfrom torch.autograd import Variablex = Variable(torch.randn(100, 100))y = Variable(torch.randn(100, 100))loos_f = torch...原创 2020-01-11 10:47:10 · 4779 阅读 · 2 评论 -
使用Pytorch的torch.nn搭建网络模型
torch.nn实现了很多与神经网络相关的类，使用这个可以很方便的搭建起模型。import torchfrom torch.autograd import Variablebatch_n = 100hidden_layer = 100input_data = 1000output_data = 10x = Variable(torch.randn(batch_n, input_dat...原创 2020-01-11 10:37:13 · 655 阅读 · 0 评论 -
pytorch程序运行结束后不释放进程
初学pytorch，在使用了自动后向传播之后，程序运行结束后不自动释放进程。如图所示：在程序中显示在资源管理器目前暂时没有找到好的办法，程序运行结束之后找到Python进程关闭…...原创 2020-01-10 23:09:38 · 2999 阅读 · 4 评论