When I convert onnx model to engine, the model always be stucked below, I can‘t find problems。
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
When I convert onnx model to engine, the model always be stucked below, I can‘t find problems。
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