bloomsburyai/question-generation

Do you have a graph_def file for the models ?

pidugusundeep opened this issue · 4 comments

I am trying out https://github.com/IntelAI/tools/tree/master/tensorflow_quantization#steps-for-fp32-optimized-frozen-graph and I would like to know if you have any .pb or .pbtxt files along with the checkpoint files which you provided in the ./models

I don't have them, but you have the code that builds the graph so you should be able to write the graph to disk.

@tomhosking may I know where exactly it is?

The model is built here, and you can then access the graph with self.model.graph, eg

tf.train.write_graph(self.model.graph, path_to_model_pb,
                     'saved_model.pb', as_text=False)

(see this stackoverflow post)

@tomhosking Thanks it's working!