I noticed that an ERROR occurred, although it didn't stop the inference progress:
Resume MLP weights from ./data/ckpt/GTA.ckpt
Resume normal model from ./data/ckpt/normal.ckpt
Using pixie as HPS Estimator
Dataset Size: 5
0%| | 0/5 [00:00<?, ?it/s]
2024-03-02 16:02:28.809516226 [W:onnxruntime:Default, onnxruntime_pybind_state.cc:515 CreateExecutionProviderInstance] Failed to create TensorrtExecutionProvider.
Please reference https://onnxruntime.ai/docs/execution-providers/TensorRT-ExecutionProvider.html#requirements to ensure all dependencies are met.
1eca7a73c3c61d9debde493de37c7d99: 0%| | 0/5 [00:06<?, ?it/s
Body Fitting --- normal: 0.089 | silhouette: 0.043 | Total: 0.132: 12%|█████████▎ | 12/100 [00:01<00:13, 6.32it/s]
1eca7a73c3c61d9debde493de37c7d99: 0%| | 0/5 [00:08<?, ?it/s]
I tried to change the onnxruntime-gpu and TensorRT's version but it didn't work.