Can anyone give some example about how to use graphsurgeon?

AI & Data Science Deep Learning (Training & Inference) TensorRT
1

Because I usually use keras to train cNN, however some operations are not supported by tensorrt, Can anybody can give some examples about how to use graphsergon, Thx

2

Hello,

I’d recommend starting here https://docs.nvidia.com/deeplearning/sdk/tensorrt-api/python_api/graphsurgeon/graphsurgeon.html

and https://docs.nvidia.com/deeplearning/sdk/tensorrt-developer-guide/index.html#extending