
PyTorch
pytorch学习相关
wydxry
红叶经霜而赤,腊梅沐雪而馨!
展开
-
RuntimeError: Input type (torch.cuda.HalfTensor) and weight type (torch.cuda.FloatTensor) should be
(16位浮点数,即float16(32位浮点数,即float32`)。要解决这个问题,你需要确保输入张量和权重张量具有相同的数据类型。原创 2025-01-09 21:26:02 · 468 阅读 · 0 评论 -
浅谈Beam Search
是一种启发式搜索算法,常用于序列生成任务(如机器翻译、文本生成、语音识别等)。它在每一步生成时,保留当前最优的 ( k ) 个候选序列(( k ) 为 beam width),而不是像贪心搜索那样只保留一个最优解。通过这种方式,它能在一定程度上避免局部最优,同时减少计算量。Beam Search 是一种高效的序列生成算法,通过保留多个候选序列,能够在保证生成质量的同时减少计算量。以上代码展示了如何使用 PyTorch 实现一个简单的 Beam Search。原创 2025-01-02 22:31:37 · 643 阅读 · 0 评论 -
网络模型中fuse()函数作用以及使用前后对比
在深度学习网络中,fuse()函数通常用于将多个层或操作融合在一起,以优化推理速度或简化模型结构。这种操作在推理阶段特别有用,因为它可以减少计算量,提高推理效率。以下是一个基于 PyTorch 的示例,展示了如何将卷积层和批量归一化层进行融合,并对比使用融合和不使用融合的效果。原创 2024-12-27 11:12:13 · 439 阅读 · 0 评论 -
浅谈nn.Identity()
是 PyTorch 中的一个模块,它的作用是直接返回输入值而不作任何修改。这个模块通常用作占位符,在不需要对输入进行任何操作时使用。的使用场景通常是在需要保持网络结构不变,但希望有条件地跳过某些层时。原创 2024-12-26 10:44:48 · 349 阅读 · 0 评论 -
@register_model 装饰器
使用 装饰器来注册模型类有以下几个优势:自动化注册:易于扩展:动态实例化:代码组织清晰:减少重复代码:以下是一个使用 装饰器注册模型的完整示例:示例说明定义模型注册装饰器:定义模型类并注册:动态创建模型实例:通过这种方式,可以实现模型的自动注册和动态实例化,提高代码的灵活性和可维护性。原创 2024-12-25 21:52:34 · 510 阅读 · 0 评论 -
Qt和Libtorch部署
通过以上步骤,你可以在Qt项目中使用libtorch进行模型的推理加速。原创 2024-09-27 14:29:52 · 1238 阅读 · 0 评论 -
PyTorch学习笔记(七)RNN
# 导入相关库函数import torchimport torch.nn as nnimport torchvisionimport torchvision.transforms as transforms# 配置设备device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')# 设置超参数sequence_length = 28input_size = 28hidden_size = 128num_la原创 2021-07-25 22:41:16 · 111 阅读 · 0 评论 -
PyTorch学习笔记(六)Residual Network
# 导入相关库函数import torchimport torch.nn as nnimport torchvisionimport torchvision.transforms as transforms# 配置设备device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')# 设置超参数num_epochs = 80batch_size = 100learning_rate = 0.001# 图像预原创 2021-07-23 21:53:55 · 189 阅读 · 0 评论 -
PyTorch学习笔记(五)卷积神经网络
# 导入相关库函数import torchimport torch.nn as nnimport torchvisionimport torchvision.transforms as transformsimport time# 配置设备device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')# 设置超参数num_epochs = 10num_classes = 10batch_size = 200原创 2021-07-23 21:51:02 · 265 阅读 · 0 评论 -
PyTorch学习笔记(四)Logistic回归
# 导入相关库函数import numpy as npimport torchimport torch.nn as nnimport torchvisionimport torchvision.transforms as transforms# 超参数input_size =28*28num_classes =10num_epochs = 10batch_size = 100learning_rate = 0.001# MNIST数据集train_dataset = torch原创 2021-07-22 22:56:19 · 105 阅读 · 0 评论 -
PyTorch学习笔记(三)线性回归模型
# 导入相关库函数import torchimport torchvision.transforms as transformsimport numpy as npimport torch.nn as nnimport matplotlib.pyplot as plt# 超参数input_size = 1output_size = 1num_epochs = 60learning_rate = 0.001# 数据集x_train = np.array([[3.3], [4.4],原创 2021-07-22 22:01:01 · 93 阅读 · 0 评论 -
PyTorch学习笔记(二)前馈神经网络
# 导入相关库函数import torchimport torch.nn as nnimport torchvisionimport torchvision.transforms as transforms# 设备配置device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')# 超参数input_size = 784hidden_size = 500num_classes = 10num_epochs原创 2021-07-22 19:55:04 · 538 阅读 · 0 评论 -
PyTorch学习笔记(一)
import torch import torchvisionimport torch.nn as nnimport torchvision.transforms as transformsimport numpy as npBasic autograd example 1# 创建tensorsx = torch.tensor(1.,requires_grad=True)w = torch.tensor(2.,requires_grad=True)b = torch.tensor(3原创 2021-07-22 19:09:07 · 214 阅读 · 0 评论