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