pytorch速成

有其他框架的基础,看一下这个项目中的速成部分,能快速上手
https://github.com/ceo1207/pytorch-lesson-zh

新建网络

A、使用tensor运算,会自动构建网络
tensor运算
可以使用运算符,+、-等,可以使用torch.xx,也可以tensor.xx在原tensor上做运算
Tensor
sum(dim=None, keepdim=False, dtype=None) → Tensor
pow(exponent) → Tensor
mm(mat2) → Tensor

torch
torch.mm(mat1, mat2, out=None) → Tensor
torch.clamp(input, min, max, out=None) → Tensor

B、override forward and init
继承Module,复写相关函数,之后讲述

如何向网络输入,获取输出和loss,更新权重

可以全手动计算grad、更新参数,这对新手理解神经网络的传播和反向传播很有帮助

示例:

import torch
# M是样本数量,input_size是输入层大小
# hidden_size是隐含层大小,output_size是输出层大小
M, input_size, hidden_size, output_size = 64, 1000, 100, 10

# 生成随机数当作样本
x = torch.randn(M, input_size) #size(64, 1000)
y = torch.randn(M, output_size) #size(64, 10)

# 参数初始化
def init_parameters():
    w1 = torch.randn(input_size, hidden_size)
    w2 = torch.randn(hidden_size, output_size)
    b1 = torch.randn(1, hidden_size)
    b2 = torch.randn(1, output_size)
    return {"w1": w1, "w2":w2, "b1": b1, "b2": b2}

# 定义模型
def model(x, parameters):
   
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
对于深度学习速成PyTorch是一个非常强大且流行的工具。下面是一些可以帮助您快速入门PyTorch的步骤: 1. 安装PyTorch:首先,您需要在您的计算机上安装PyTorch。您可以从官方网站(https://pytorch.org/)上找到适合您操作系统和硬件配置的安装指南。 2. 学习基本概念:在开始使用PyTorch之前,了解一些基本的深度学习概念是很重要的。这包括理解神经网络、损失函数、优化算法等。您可以阅读一些在线教程或参考书籍来学习这些概念。 3. 构建简单的模型:使用PyTorch构建一个简单的模型是入门的好方法。您可以从一些经典问题开始,例如手写数字识别(MNIST数据集)或图像分类(CIFAR-10数据集)。通过实际动手构建模型,您将更好地理解PyTorch的工作原理。 4. 使用预训练模型:PyTorch提供了许多预训练模型,这些模型在各种任务上已经取得了良好的效果。您可以尝试使用这些预训练模型来进行图像分类、目标检测、自然语言处理等任务。这将帮助您快速获得令人满意的结果。 5. 深入学习:一旦您掌握了基本的PyTorch知识,可以深入学习更高级的主题。这包括使用自定义数据集、使用GPU加速训练、处理序列数据等。还有许多优秀的在线资源和教程可供参考。 希望这些步骤可以帮助您快速入门PyTorch深度学习!如果您有任何问题,请随时问我。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值