search

bloomsburyai / question-generation

Neural text-to-text question generation
MIT License
217 stars 52 forks source link

Do you have a graph_def file for the models ? #27

Closed pidugusundeep closed 5 years ago

pidugusundeep commented 5 years ago

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

tomhosking commented 5 years ago

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.

pidugusundeep commented 5 years ago

@tomhosking may I know where exactly it is?

tomhosking commented 5 years ago

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)

pidugusundeep commented 5 years ago

@tomhosking Thanks it's working!