![](https://img-blog.csdnimg.cn/20201014180756738.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
文章平均质量分 67
这不是荒野
自动驾驶感知工程师
展开
-
pytorch自定义算子并导出onnx计算图详细代码教程
pytorch自定义算子例子,本案例中包含了多输入多输出,以及张量权重、不同类型attributes的设置方法。原创 2022-12-09 21:34:29 · 2499 阅读 · 2 评论 -
【PYTORCH >> ONNX >> TENSORRT】自定义plug-in算子解析onnx模型生成引擎详细步骤
pytorch >> onnx >> tensorrt 自定义plugin生成engine关键步骤记录。原创 2022-12-03 14:07:46 · 2138 阅读 · 2 评论 -
实验小测 BatchNorm2D 层对梯度传导的影响 (batchnorm层减去均值和除以标准差这两个步骤所使用的tensor是否detach)
如今BatchNorm层已经广泛被应用于各种神经网络内,为了更加深入的了解该层的输出结果反向传播梯度的细节,提出问题:batchnorm层减去均值和除以标准差这两个步骤所使用的tensor是否做了detach处理?做了简单的测试,测试过程如下: 首先简历好实验所需数据,图片的batchsize=2,其CHW三个属性都为原创 2022-03-28 14:54:48 · 873 阅读 · 0 评论 -
梯度反向传播过程中,测试强行修改后的预测结果是否还会传递loss?
pytorch反向传播梯度细节问题原创 2022-03-25 14:34:24 · 1922 阅读 · 0 评论