load .pt file error engine_bytes = state_dict[prefix + "engine"] KeyError: 'engine'

NVIDIA-AI-IOT / trt_pose_hand

Real-time hand pose estimation and gesture classification using TensorRT

MIT License

206 stars 49 forks source link

Open 2443374695 opened 3 months ago

2443374695 commented 3 months ago

model_trt.load_state_dict(torch.load('epoch20.pt')) engine_bytes = state_dict[prefix + "engine"] KeyError: 'engine'

aditya-nisal commented 2 months ago

Hello, were you able to fix the error? I am running into the same error.