Pytorch
六甲横宝
这个作者很懒,什么都没留下…
展开
-
Pytorch 03 线性回归实例
import torch from torch.nn import Linear, Module, MSELoss from torch.optim import SGD import numpy as np import pandas as pd import matplotlib.pyplot as plt import seaborn as sns # 定义一个线性函数,这里使用y=5...原创 2019-07-30 10:58:13 · 183 阅读 · 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 · 440 阅读 · 0 评论 -
Pytorch 02 一个简单的分类器
import torch import torchvision import torchvision.transforms as transforms import matplotlib.pyplot as plt import numpy as np import torch.nn as nn import torch.nn.functional as F import torch.optim...原创 2019-07-27 11:21:39 · 499 阅读 · 0 评论 -
Pytorch 01 Pytorch安装及环境配置
需要注意的是需要先安装好CUDA(安装如下),然后再根据相应CUDA版本选择Pytorch相应版本安装。 Pytorch官网https://pytorch.org/中查看相关版本及对应系统,在命令行中输入"Run this Command"中代码即可 其中需要注意的是自己电脑的CUDA版本,CUDA版本必须对应,否则Pytorch会运行出错,其中CUDA安装过程如下: 首先需要查看自己电...原创 2019-07-27 11:14:32 · 2282 阅读 · 0 评论