/keras_to_trt

Sample of Model training in Keras, then conversion to TensorRT

Primary LanguageJupyter NotebookMIT LicenseMIT

Keras to TensorRT

This is sample notebook showing how to convert Keras model into TensorRT

  • Work is in progress
  • Code was originally created for TensorFlow version 1.14 (with built-in Keras library)