Search
 
SCRIPT & CODE EXAMPLE
 
CODE EXAMPLE FOR PYTHON

RuntimeError: Please use tf.experimental.tensorrt.Converter in TF 2.0. site:stackoverflow.com

params = trt.DEFAULT_TRT_CONVERSION_PARAMS._replace(
    precision_mode='FP16',
    is_dynamic_op=True)
    
converter = trt.TrtGraphConverterV2(
    input_saved_model_dir=saved_model_dir,
    conversion_params=params)
converter.convert()
saved_model_dir_trt = "/tmp/model.trt"
converter.save(saved_model_dir_trt)
Source by stackoverflow.com #
 
PREVIOUS NEXT
Tagged: #Please #TF
ADD COMMENT
Topic
Name
3+5 =