-
Notifications
You must be signed in to change notification settings - Fork 8
Open
Description
I did a custom train at Yolov7
asert(engine->getNbBindings() == 2);
In this code, getNbBindings() has a value of 5
Are there any issues I need to pay attention to in train?
'''
[W] [TRT] CUDA lazy loading is not enabled. Enabling it can significantly reduce device memory usage. See CUDA_MODULE_LOADING in https://docs.nvidia.com/cuda/cuda-c-programming-guide/index.html#env-vars
alta-yolov7-deepsort: ../alta-yolov7-deepsort/model.cpp:96: void Model::LoadEngine(): Assertion `engine->getNbBindings() == 2' failed.
'''
Metadata
Metadata
Assignees
Labels
No labels