人工智能
-
PyTorch保存、加载模型，PyTorch中已封装的网络模型
state_dict()函数可以返回所有的状态数据。load_state_dict()函数可以加载这些状态数据。 推荐使用： #保存 t.save(net.state_dict(),”net.pth”) #加载 net2=Net() net2.load_state_dict(t.load(“net.pth”)) 不推荐直接save与load，因为这种方式严重…
-
Pytorch快速入门及在线体验
本文搭配了Pytorch在线环境，可以直接在线体验。 Pytorch是Facebook 的 AI 研究团队发布了一个基于 Python的科学计算包，旨在服务两类场合： 1.替代numpy发挥GPU潜能 ；2. 一个提供了高度灵活性和效率的深度学习实验性平台。 1.Pytorch简介 Pytorch是Facebook 的 AI 研究团队发布了一个基于 Pyth…
-
pytorch 2 variable 变量
import torch from torch.autograd import Variable tensor = torch.FloatTensor([[1, 2], [3, 4]]) variable = Variable(tensor, requires_grad=True) print(tensor) # tensor([[1., 2.], [3.,…
-
pytorch练习
1、使用梯度下降法拟合y = sin(x) import numpy as np import torch import torchvision import torch.optim as optim import torch.nn as nn import torch.nn.functional as F import time import os fro…
-
全面解析Pytorch框架下模型存储，加载以及冻结
最近在做试验中遇到了一些深度网络模型加载以及存储的问题，因此整理了一份比较全面的在 PyTorch 框架下有关模型的问题。首先咱们先定义一个网络来进行后续的分析： 1、本文通用的网络模型 import torch import torch.nn as nn ”’ 定义网络中第一个网络模块 Net1 ”’ class Net1(nn.Module): de…
-
Linux下conda配置虚拟环境：python + pytorch
Linux下conda配置虚拟环境：python + pytorch 默认已经安装好conda 创建虚拟环境 conda创建并激活虚拟环境 命令： conda create -n your_env_name python=2.7/3.6source activate your_env_name 其中，-n中n表示name，即你创建环境的名字。之后如果忘记自己…
-
pytorch tensor 维度理解.md
torch.randn torch.randn(*sizes, out=None) → Tensor(张量) 返回一个张量，包含了从标准正态分布(均值为0，方差为 1)中抽取一组随机数，形状由可变参数sizes定义。 参数: sizes (int…) – 整数序列，定义了输出形状 out (Tensor, optinal) – 结果张量 二维 >&…
-
pytorch 图片处理.md
本篇所有代码位置链接???? pytorch 图片处理，主要用到 torchvision 模块的 datasets 和 transforms。 例如：本地图片资源目录结构如下 ➜ torch_test tree animal_data animal_data ├── train │ ├── ants │ │ ├── 0013035.jpg │ …
-
Pytorch多GPU训练
临近放假, 服务器上的GPU好多空闲, 博主顺便研究了一下如何用多卡同时训练 原理 多卡训练的基本过程 首先把模型加载到一个主设备 把模型只读复制到多个设备 把大的batch数据也等分到不同的设备 最后将所有设备计算得到的梯度合并更新主设备上的模型参数 代码实现(以Minist为例) #!/usr/bin/python3 # coding: utf-8 im…
-
pytorch加载和保存模型
在模型完成训练后，我们需要将训练好的模型保存为一个文件供测试使用，或者因为一些原因我们需要继续之前的状态训练之前保存的模型，那么如何在PyTorch中保存和恢复模型呢？ 方法一(推荐)： 第一种方法也是官方推荐的方法，只保存和恢复模型中的参数。 保存 torch.save(the_model.state_dict(), PATH) 恢复 the_mo…
