# onnx_debug_tool
**Repository Path**: evan-q/onnx_debug_tool
## Basic Information
- **Project Name**: onnx_debug_tool
- **Description**: No description available
- **Primary Language**: Python
- **License**: Apache-2.0
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No
## Statistics
- **Stars**: 0
- **Forks**: 0
- **Created**: 2022-02-25
- **Last Updated**: 2023-04-03
## Categories & Tags
**Categories**: Uncategorized
**Tags**: None
## README
# onnx_debug_tool
功能：
----
调试ONNX模型，自动加载运行模型两次(cpu、gpu)，对比cpu与gpu的执行结果，不一致时输出错误节点的名称及输入信息。
运行方式：
--------
python onnx_debug.py -i ./test.onnx
当前限制：
---------
1 输入数据随机生成，暂不支持读取pb文件作为输入。
2 暂未在gpu环境测试，只在cpu环境下模拟了output不一致的场景，如需测试gpu，请在代码中搜索EP_list进行替换。
输出：
------------
1 结果不一致
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
WARNING: output 2 is abnormal, please check it~~
Dismatch node name: MatMul_1 , input: ['0', '1']
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
2 结果一致
=================================================
=================================================
Congratulations, it works well as expected