• S
    swisspotato

    [02/12/2021-20:39:55] [V] [TRT] 876:Constant ->
    [02/12/2021-20:39:55] [V] [TRT] 877:Mul -> (3, 40, 40, 2)
    [02/12/2021-20:39:55] [V] [TRT] 878:Constant ->
    Unsupported ONNX data type: DOUBLE (11)
    [02/12/2021-20:39:55] [V] [TRT] 879:Sub -> (3, 40, 40, 2)
    [02/12/2021-20:39:55] [V] [TRT] 880:Constant -> (1, 1, 40, 40, 2)
    [02/12/2021-20:39:55] [V] [TRT] 881:Add -> (3, 40, 40, 2)
    [02/12/2021-20:39:55] [V] [TRT] 882:Constant ->
    [02/12/2021-20:39:55] [V] [TRT] 883:Mul -> (3, 40, 40, 2)
    While parsing node number 438 [Cast -> "884"]:
    --- Begin node ---
    input: "883"
    output: "884"
    op_type: "Cast"
    attribute {
    name: "to"
    i: 1
    type: INT
    }

    --- End node ---
    ERROR: builtin_op_importers.cpp:727 In function importCast:
    [8] Assertion failed: trt_dtype == nvinfer1::DataType::kHALF && cast_dtype == ::ONNX_NAMESPACE::TensorProto::FLOAT
    [02/12/2021-20:39:55] [E] Failed to parse onnx file
    [02/12/2021-20:39:55] [E] Parsing model failed
    [02/12/2021-20:39:55] [E] Engine could not be created
    &&&& FAILED TensorRT.trtexec # ./trtexec --onnx=/home/aa/onnx_trt_models/best_float32.onnx --verbose

    大佬们有遇到这个问题吗?使用 onnx-tensorrt转换trt模型失败,求助

    发布在 社区求助区(SOS!!) 阅读更多

与 神力AI社区 的连接断开,我们正在尝试重连,请耐心等待