![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch
六甲横宝
这个作者很懒,什么都没留下…
展开
-
Pytorch 03 线性回归实例
import torchfrom torch.nn import Linear, Module, MSELossfrom torch.optim import SGDimport numpy as npimport pandas as pdimport matplotlib.pyplot as pltimport seaborn as sns# 定义一个线性函数,这里使用y=5...原创 2019-07-30 10:58:13 · 207 阅读 · 0 评论 -
Pytorch 04 激活函数
ax = plt.gca()ax.spines['right'].set_color('none')ax.spines['top'].set_color('none')ax.xaxis.set_ticks_position('bottom')ax.spines['bottom'].set_position(('data', 0))ax.yaxis.set_ticks_positio...原创 2019-07-30 11:18:53 · 450 阅读 · 0 评论 -
Pytorch 02 一个简单的分类器
import torchimport torchvisionimport torchvision.transforms as transformsimport matplotlib.pyplot as pltimport numpy as npimport torch.nn as nnimport torch.nn.functional as Fimport torch.optim...原创 2019-07-27 11:21:39 · 505 阅读 · 0 评论 -
Pytorch 01 Pytorch安装及环境配置
需要注意的是需要先安装好CUDA(安装如下),然后再根据相应CUDA版本选择Pytorch相应版本安装。Pytorch官网https://pytorch.org/中查看相关版本及对应系统,在命令行中输入"Run this Command"中代码即可其中需要注意的是自己电脑的CUDA版本,CUDA版本必须对应,否则Pytorch会运行出错,其中CUDA安装过程如下:首先需要查看自己电...原创 2019-07-27 11:14:32 · 2287 阅读 · 0 评论