search

ardianumam / Tensorflow-TensorRT

This repository is for my YT video series about optimizing a Tensorflow deep learning model using TensorRT. We demonstrate optimizing LeNet-like model and YOLOv3 model, and get 3.7x and 1.5x faster for the former and the latter, respectively, compared to the original models.
303 stars 110 forks source link

Tensorflow to Frozen model error #29

Closed hidden1001 closed 4 years ago

hidden1001 commented 4 years ago

Hello, i am using Linux based Jetson nano and when trying to convert my tensorflow model to Frozen model i am getting this error:

assert d in name_to_node, "%s is not in graph" % d AssertionError: output_tensor/softmax is not in graph

hidden1001 commented 4 years ago

Sorry guys, I found the solution model = keras.models.load_model( 'model.h5' ) output = model.output print(output )