smellslikeml / ActionAI

Real-Time Spatio-Temporally Localized Activity Detection by Tracking Body Keypoints
https://www.hackster.io/actionai/actionai-custom-tracking-multiperson-activity-recognition-fa5cb5
GNU General Public License v3.0
765 stars 191 forks source link

Which version of TensorRT is used for this project #68

Open Have-a-wind opened 1 year ago

Have-a-wind commented 1 year ago

When I try to run iva.py, I get an error, and the specific error returned is [05/23/2023-08:41:34] [TRT] [E] 1: [stdArchiveReader.cpp::StdArchiveReader::30] Error Code 1: Serialization (Serialization assertion magicTagRead == magicTag failed.Magic tag does not match) [05/23/2023-08:41:34] [TRT] [E] 4: [runtime.cpp::deserializeCudaEngine::50] Error Code 4: Internal Error (Engine deserialization failed.) Traceback (most recent call last): File "iva.py", line 8, in <module> from model import * File "/home/wheeltec/ActionAI/model.py", line 73, in <module> model_trt.load_state_dict(torch.load(MODEL_PATH)) File "/home/wheeltec/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 1037, in load_state_dict load(self) File "/home/wheeltec/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 1032, in load state_dict, prefix, local_metadata, True, missing_keys, unexpected_keys, error_msgs) File "/usr/local/lib/python3.6/dist-packages/torch2trt-0.4.0-py3.6-linux-aarch64.egg/torch2trt/torch2trt.py", line 591, in _load_from_state_dict self.context = self.engine.create_execution_context() AttributeError: 'NoneType' object has no attribute 'create_execution_context'