PyTorch
samylee
机器之眼,看到你看不到的!
展开
-
PyTorch分类网络:Python训练_测试_模型转换 && Windows_LibTorch_C++部署
第一部分 前言众位小伙伴,好久没更新博客了,本次为大家带来系列博客:如何将PyTorch训练的网络通过模型转换,部署到Windows纯C++下执行,落地应用。这里并没有将PyTorch模型转至其他深度学习框架下,而是通过PyTorch的LibTorch来完成相关C++的部署应用。第二部分 说明PyTorch版本:Torch-1.4.0-cu101LibTorch版本:LibTorch-1.4.0-cu101Anaconda版本:Anaconda3-Python3.6GPU:GTX10原创 2020-05-25 15:50:47 · 1391 阅读 · 0 评论 -
PyTorch-PyramidBoxDemo
PyTorch-PyramidBoxDemo硬件:NVIDIA-GTX1080软件:Windows10、python3.6.5、pytorch-gpu-0.4.1一、基础知识(一图看懂)二、代码展示1、demo.py#-*- coding:utf-8 -*-from __future__ import divisionfrom __future__ imp...原创 2019-07-09 17:55:50 · 996 阅读 · 2 评论 -
PyTorch-GPU加速
PyTorch-GPU加速硬件:NVIDIA-GTX1080软件:Windows7、python3.6.5、pytorch-gpu-0.4.1一、基础知识将数据和网络都推到GPU,接上.cuda()二、代码展示import torchimport torch.nn as nnimport torch.utils.data as Dataimport torchvis...原创 2019-06-12 13:35:03 · 9982 阅读 · 2 评论 -
PyTorch-数据批处理
PyTorch-数据批处理硬件:NVIDIA-GTX1080软件:Windows7、python3.6.5、pytorch-gpu-0.4.1一、基础知识1、学习TensorDataset2、学习DataLoader二、代码展示import torchimport torch.utils.data as Datatorch.manual_seed(1) #...原创 2019-06-04 15:25:37 · 1203 阅读 · 0 评论 -
PyTorch-模型保存与提取
PyTorch-模型保存与提取硬件:NVIDIA-GTX1080软件:Windows7、python3.6.5、pytorch-gpu-0.4.1一、基础知识1、save有两种形式,一种是全图保存,一种是参数保存2、load也有两种形式,一种是全图加载,无需单独建立模型,一种是参数加载,但是需要单独建立模型二、代码展示import torchimport matpl...原创 2019-06-03 15:43:35 · 1356 阅读 · 0 评论 -
PyTorch-Sequential网络快速搭建法
PyTorch-Sequential网络快速搭建法硬件:NVIDIA-GTX1080软件:Windows7、python3.6.5、pytorch-gpu-0.4.1一、基础知识1、torch.nn.Sequential()方法与class Net方法有相同功效,且Sequential方法更简单明了!2、以Linear关系分类为例(https://blog.csdn.net/s...原创 2019-06-01 17:37:52 · 703 阅读 · 0 评论 -
PyTorch-Activation激活函数
PyTorch-Activation激活函数硬件:NVIDIA-GTX1080软件:Windows7、python3.6.5、pytorch-gpu-0.4.1一、基础知识1、激活函数作用:神经网络可以描述非线性问题2、relu、sigmoid、tanh、softplus二、代码展示import torchimport torch.nn.functional as ...原创 2019-05-29 15:39:49 · 2382 阅读 · 0 评论 -
PyTorch-初遇卷积神经网络
PyTorch-初遇卷积神经网络硬件:NVIDIA-GTX1080软件:Windows7、python3.6.5、pytorch-gpu-0.4.1一、基础知识MNIST数据测试分类二、代码展示import os# third-party libraryimport torchimport torch.nn as nnimport torch.utils.dat...原创 2019-06-06 14:45:02 · 1189 阅读 · 0 评论 -
PyTorch-Linear关系分类
PyTorch-Linear关系分类硬件:NVIDIA-GTX1080软件:Windows7、python3.6.5、pytorch-gpu-0.4.1一、基础知识1、二分类问题2、函数学习torch.normal()、torch.cat()、torch.nn.CrossEntropyLoss()、torch.max(Func.softmax())二、代码展示impo...原创 2019-05-31 17:26:37 · 2403 阅读 · 0 评论 -
PyTorch-Variable变量
PyTorch-Variable变量硬件:NVIDIA-GTX1080软件:Windows7、python3.6.5、pytorch-gpu-0.4.1一、基础知识variable = torch.autograd.Variable(tensor, requires_grad = False),requires_grad表示参不参与误差反向传播二、代码展示import t...原创 2019-05-28 18:04:00 · 1483 阅读 · 0 评论 -
PyTorch-加速网络训练
PyTorch-加速网络训练硬件:NVIDIA-GTX1080软件:Windows7、python3.6.5、pytorch-gpu-0.4.1一、基础知识1、SGD+Momentum2、RMSprop3、Adam二、代码展示import torchimport torch.utils.data as Dataimport torch.nn.fun...原创 2019-06-05 16:59:24 · 1169 阅读 · 0 评论 -
PyTorch-Linear关系拟合
PyTorch-Linear关系拟合硬件:NVIDIA-GTX1080软件:Windows7、python3.6.5、pytorch-gpu-0.4.1一、基础知识1、问题:y = a * x^2 + b2、继承 torch 的 Module二、代码展示import torchimport torch.nn.functional as Func # 激励函数...原创 2019-05-30 14:47:50 · 827 阅读 · 0 评论 -
PyTorch-基本数据操作(Numpy)
PyTorch-基本数据操作(Numpy)硬件:NVIDIA-GTX1080软件:Windows7、python3.6.5、pytorch-gpu-0.4.1一、基础知识1、Torch 为神经网络界的 Numpy,torch.from_numpy() 与 torch_data.numpy() 即可完成torch数据和numpy数据的相互转化2、Torch 浮点数接收方式,tor...原创 2019-05-27 16:39:31 · 4694 阅读 · 0 评论