fly520fly520

Segmentation fault (core dumped)

4

My Code : # create example data, x = torch.ones((1, 3, 112, 112)).cuda() model_trt_new = TRTModule() model_trt_new.load_state_dict(torch.load('face_trt.pth')) y_trt_new = model_trt_new(x) This code can run normally, but when it is finished,...