PyTorch
ZZ58
这个作者很懒,什么都没留下…
展开
-
Pytorch实现手写数字识别
'''用PyTorch完成手写数字识别1.准备数据2.构建模型3.模型的训练4.模型的保存5.模型的评估'''import torchimport osimport numpy as npfrom torch import nnfrom torch import optimimport torch.nn.functional as Ffrom torch.utils.data import DataLoaderfrom torchvision.datasets import原创 2021-05-09 14:41:39 · 198 阅读 · 0 评论 -
PyTorch实现线性回归
import torchimport torch.nn as nnfrom torch.optim import SGDimport matplotlib.pyplot as plt#0.准备数据x = torch.rand([500,1])y_true = 3*x + 0.8#1.定义模型class MyLinear(nn.Module): def __init__(self): super(MyLinear, self).__init__() se原创 2021-05-09 10:53:47 · 74 阅读 · 0 评论 -
Pytorch--常用API
1.nn.Module a.__init__, b.farward nn.Module定义了__call__方法,及Lr的实例,能够直接被传入参数调用,实际上用的是forward方法并传入参数from torch import nnclass Lr(nn.Module): def __init__(self): super(Lr,self).__init_() self.linear = nn.Linear(1,1) def forward(self,x): out = se原创 2021-05-07 14:37:31 · 194 阅读 · 0 评论 -
安装PyTorch以及GPU上运行
Pytorch:torch,torchvision,cudatoolkit 对应的版本https://pytorch.org/get-started/previous-versions/下载torch以及torchvision,toolkit网址http://download.pytorch.org/whl/torch_stable.html下载下来的whl文件在这个目录下pip install xxxxxx.whl在GPU上执行程序a.自定义的参数需要转化为cuda支持的tensorb.原创 2021-05-07 15:15:11 · 62 阅读 · 0 评论