Pytorch

15篇文章
1 我的深度学习项目代码文件组织结构 一般来说,深度学习项目需要包含以下内容: 数据集预处理与加载 深度学习模型定义 模型训练 模型推理 根据以上的功能描述,我的深度学习项目代码文件组织结构如下: ├…
1 Pytorch的随机种子 最近在看一些开源的Pytorch项目时,几乎每一个项目都会设置随机数种子,比如下面这种 class RandomState(object): def __init__(s…
1 torch.nn.Conv1d torch.nn.Conv1d主要是对一维输入Tensor应用一维卷积。 如果一维卷积输入为(N,C_{in},L),输出为(N,C_{out},L_{out}),…
1 torchsummary/torchsummaryX torchsummary Github地址:https://github.com/sksq96/pytorch-summary torchsu…
1 torch.nn.LSTM torch.nn.LSTM是pytorch内置的LSTM模块。 对于torch.nn.LSTM输入序列的每一个元素,都使用以下经典的LSTM计算过程: \begin{a…
CTC(Connectionist Temporal Classification)主要是处理不定长序列对齐问题,而CTCLoss主要是计算连续未分段的时间序列与目标序列之间的损失。CTCLoss对输…
1 transpose和permute函数的区别和用法 Pytorch中的transpose和permute方法都可以实现tensor维度之间的变换,transpose一次只能转换两个维度,而perm…
在本文中,我将介绍如何使用普通算子matmul对torch.einsum('nkctv,kvw->nctw',(a,b))的算子模式进行替代。之前在另一篇文章:https://www.stubborn…
在本文中,我将介绍如何使用普通算子matmul对torch.einsum('bhxyd,md->bhxym',(a,b))的算子模式进行替代。之前在另一篇文章:https://www.stubborn…
1 pytorch的torch.matmul()函数 函数形式 torch.matmul(input, other, *, out=None) → Tensor 该函数主要是用于求解两个tensor的…
使用Pytorch从零训练一个深度学习网络模型,常需要经过以下步骤:自定义数据集、加载自定义数据集、网络模型结构定义、定义损失函数、定义优化器、训练模型、测试模型、保存与加载模型等步骤。下文将详细阐述…
1 torch.onnx.export torch.onnx.export( model, args, f, export_params=True, verbose=False, training=&…

关注我们的公众号

微信公众号