pytorch
bineleanor
这个作者很懒,什么都没留下…
展开
-
Pytorch:nn, optim, nn Modules
参考资料:https://pytorch.org/tutorials/beginner/pytorch_with_examples.htmlPytorch:nnpytorch的nn包中定义了一系列的Modules, 大致相当于神经网络层,Module接收输入张量以及计算输出张量,内部状态也可以是可学习的张量形式。import torchN, D_in, H, D_out = 64, 10...原创 2019-04-06 10:47:46 · 284 阅读 · 0 评论 -
pytorch:neural network
参考链接:https://pytorch.org/tutorials/beginner/blitz/neural_networks_tutorial.html#sphx-glr-beginner-blitz-neural-networks-tutorial-py一个典型的神经网络的训练过程可以描述如下:1.定义神经网络(包含一些可以训练的参数);2.根据输入的数据集进行迭代;3.通过网络...原创 2019-04-06 17:24:25 · 266 阅读 · 0 评论 -
pytorch:对CIFAR10图片进行分类
参考资料:https://pytorch.org/tutorials/beginner/blitz/cifar10_tutorial.html#sphx-glr-beginner-blitz-cifar10-tutorial-py图片分类模型的具体步骤:1.加载数据,对数据进行归一化处理(使用torchvision)2.定义卷积神经网络3.定义损失函数4.在训练集上训练神经网络5.在测...原创 2019-04-06 18:00:15 · 1258 阅读 · 0 评论 -
pytorch在CPU和GPU上加载模型
参考博客:https://blog.csdn.net/dcrmg/article/details/79503978CPU训练的模型在CPU上加载: CPU —> CPUtorch.load(model_path)GPU训练的模型在GPU上加载: GPU—> GPUtorch.load(model_path)CPU训练的模型在GPU上加载: CPU—> G...转载 2019-04-10 21:19:35 · 1766 阅读 · 0 评论 -
Pytorch:Tensor
参考资料:https://pytorch.org/tutorials/beginner/pytorch_with_examples.htmlNumpy是一个强有力的框架,但是不能使用GPU进行数值计算,故numpy对于现有的深度学习很多时候使用起来是不足够的。Pytorch的一个基础概念:张量在概念上pytorch的张量等同于numpy array,张量可以看作n维的数组,pytorch提供...原创 2019-04-01 21:37:48 · 346 阅读 · 0 评论 -
Pytorch:Defining new autograd functions
参考资料:https://pytorch.org/tutorials/beginner/pytorch_with_examples.html前向函数:根据输入的张量计算输出的张量值;反向函数:接受输出的张量,计算关于输入张量的梯度。在pytorch中可以很容易地通过定义torch.autograd.Function的子类定义自动求梯度的算子,并且实现前向和反向的函数,可以通过我们定义的自动求...原创 2019-04-01 22:07:02 · 358 阅读 · 0 评论