Skip to content

onnx to trt errors #235

@jiamfeng

Description

@jiamfeng

When I convert onnx model to engine, the model always be stucked below, I can‘t find problems。

Image

to onnx command:
XFORMERS_DISABLED=1 python scripts/make_onnx.py --save_path ./output/foundation_stereo.onnx --ckpt_dir ./pretrained_models/11-33-40/model_best_bp2.pth --height 448 --width 768 --valid_iters 20
to trt:

trtexec --onnx=output/foundation_stereo.onnx --verbose --saveEngine=output/foundation_stereo.engine --fp16

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions