tf_trt_models icon indicating copy to clipboard operation
tf_trt_models copied to clipboard

Published 20 hours ago • verified publisher icon NVIDIA-AI-IOT

tensorflow 2.x

Open drakkan opened this issue 5 years ago • 2 comments

Please add support for tensorflow 2.x, thanks!

drakkan avatar Dec 02 '19 10:12 drakkan

Seconded. I get many issues with converting TF2 models to uff/frozen graph, ..

rvorias avatar Dec 20 '19 11:12 rvorias

tensorflow 2.1 include a conversion API to trt (tf.experimental.tensorrt.Converter)

https://github.com/tensorflow/tensorflow/releases/tag/v2.1.0-rc0

we probably should test this converter

drakkan avatar Dec 20 '19 11:12 drakkan