Open 2443374695 opened 3 months ago
model_trt.load_state_dict(torch.load('epoch20.pt')) engine_bytes = state_dict[prefix + "engine"] KeyError: 'engine'
Hello, were you able to fix the error? I am running into the same error.
model_trt.load_state_dict(torch.load('epoch20.pt')) engine_bytes = state_dict[prefix + "engine"] KeyError: 'engine'