-
Notifications
You must be signed in to change notification settings - Fork 372
Closed
Labels
bugSomething isn't workingSomething isn't workingrelease: patchThis change needs to go out as a patch to the current versionThis change needs to go out as a patch to the current version
Description
Bug Description
Got this error:
INFO: [Torch-TensorRT - Debug Build] - [MemUsageChange] TensorRT-managed allocation in IExecutionContext creation: CPU +0, GPU +132, now: CPU 0, GPU 540 (MiB)
terminate called after throwing an instance of 'std::invalid_argument'
what(): stoi
Aborted (core dumped)
Debugged and found that it fails at this line: https://github.com/NVIDIA/Torch-TensorRT/blob/c95229144432d96f4bdaa71fb1d242242d42bc29/core/runtime/TRTEngine.cpp#L63
To Reproduce
MaskRCNN model from detectron2.
Expected behavior
should figure out why we cannot find a number from this engine.
Metadata
Metadata
Assignees
Labels
bugSomething isn't workingSomething isn't workingrelease: patchThis change needs to go out as a patch to the current versionThis change needs to go out as a patch to the current version