Tensorflow-TensorRT
Tensorflow-TensorRT copied to clipboard
Segmentation Fault while creating TRT_GRAPH
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?
I got the same issue with the code running in TX2