Tensorflow-TensorRT icon indicating copy to clipboard operation
Tensorflow-TensorRT copied to clipboard

Segmentation Fault while creating TRT_GRAPH

Open VibAltekar opened this issue 6 years ago • 1 comments

When I run ipynb 7, and specifically

your_outputs = ["Placeholder:0", "concat_9:0", "mul_9:0"] trt_graph = trt.create_inference_graph( input_graph_def=frozen_graph,# frozen model outputs=your_outputs, max_batch_size=1,# specify your max batch size max_workspace_size_bytes=2*(10**9),# specify the max workspace precision_mode="FP16") # precision, can be "FP32" (32 floating point precision) or "FP16"

and I get a segmentation fault. -> [1] 20875 segmentation fault (core dumped) sudo python3 yoloconvert.py

Anyone else run into this problem?

VibAltekar avatar Jul 25 '19 10:07 VibAltekar

I got the same issue with the code running in TX2

WangboML avatar Aug 22 '19 12:08 WangboML