- 博客(4)
- 问答 (1)
- 收藏
- 关注
RSS订阅原创 torch.bitwise_right_shift和bitwise_and导出到onnx算子的问题
代码中使用了bitpack来量化-反量化，其中用到了torch.bitwise_right_shift和bitwise_and函数，在导出ONNX模型时报错，特此记录。
2023-07-09 15:42:13
897
原创 使用Bolt框架部署二值化姿态估计模型
一、二值姿态估计网络二值网络较难用后训练的方式进行量化，因此本文基于Simple Baselines 提出的模型结构，替换其backbone中的卷积模块为二值卷积（其中实现参考了这篇论文）并使用pytorch框架进行训练。由于二值网络的权重在训练时仍然是浮点类型，因此在模型转换时需要将pytorch模型中的weight使用torch.sign进行二值化。随后将pytorch模型导出为onnx，再使用bolt提供的转换工具进行转换。二、实验准备实验使用的开发板为昇腾的Atlas200，CPU为8*Co
2022-05-19 11:55:32
641
2
空空如也
caffe如何在模型中预置一些要用到的矩阵？
2020-12-04
TA创建的收藏夹 TA关注的收藏夹
TA关注的人