专栏算法工具链用netron可视化生成的model.pt文件，不支持

用netron可视化生成的model.pt文件，不支持

舟行千里2023-08-11
35
4

训练/open_explorer/ddk/samples/ai_toolchain/horizon_model_train_sample/scripts下的densetnt模型，在

tmp_models/densetnt_argoverse1/compile下生成model.pt文件，尝试用netron可视化model.pt，报错：

Python Error Unknown function 'ops.horizon.convert_conv_params'.

有没有相关工具能够可视化这个图结构？

算法工具链
评论1
0/1000
  • 颜值即正义
    Lv.2

    您好，可以使用qat中的visualize_model接口来保存并查看模型的图结构，参考手册：https://developer.horizon.ai/api/v1/fileData/horizon_j5_open_explorer_cn_doc/plugin/source/quick_start/quick_start.html#%E6%A8%A1%E5%9E%8B%E9%83%A8%E7%BD%B2

    2023-08-11
    0
    3
    • 颜值即正义回复颜值即正义:
      您好，地平线工具链在持续迭代优化，为了给您提供更好的服务，希望您能抽出3分钟左右的时间，将您在使用工具链期间的感受和建议告诉我们，您的宝贵意见对我们很重要，非常感谢！
      问卷链接：地平线算法工具链使用满意度反馈
      2023-08-11
      0
    • 舟行千里回复颜值即正义:
      python3 tools/train.py --stage "qat" --config configs/traj_pred/densetnt_argoverse1.py
      执行这个命令会生成model.svg文件吗？
      2023-08-14
      0
    • 颜值即正义回复舟行千里:
      不是的，以下为正确步骤：
      1.在tools/compile_perf.py中添加可视化的代码：

      注：1. 使用可视化接口前要from horizon_plugin_pytorch.quantization import visualize_model；2.可以通过修改save_path配置可视化图存储路径；

      2. 然后再运行python3 tools/compile_perf.py --config configs/traj_pred/densetnt_argoverse1.py

      2023-08-14
      0