J5 在QAT过程中,check_model

应用开发 算法工具链
1

当有两个tensor输入时，目前的设置为check_model(script_model, [(x1,x2)])

会报错ValueError: (‘parsing prim node’, %1248 : (Tensor, Tensor) = prim::CallMethod[name=“forward”]

想请教一下是不是多个输入形式不合理。

2

你好，check_model的example_inputs的输入要求如下，建议按照example_inputs=[{‘input_name1’:tensor1,‘input_name2’:tensor2}] 的形式来构建输入：

3

还有就是，你的horizon_plugin_pytorch和hbdk的版本号是多少

4

此问题解决了 现在是compile_model时 由于grid_sample的输入为(6，C,H,W)报错 RuntimeError: HBDK model compilation FAIL, please check with HBDK Group

5

horizon_plugin_pytorch 为1.3.3 hbdk为3.42.2

6

想用最新版本的docker，里面的cuda编译器被删除了不够方便

7

麻烦发一下完整的报错截图

8

麻烦提供一下编译报错的完整截图、模型输入信息和hbir文件