search

choderalab / gimlet

Graph Inference on MoLEcular Topology
Other
26 stars 9 forks source link

Saving the model #6

Open gaurav-tikhe opened 3 years ago

gaurav-tikhe commented 3 years ago

Hello Sir/Madam, I am student of mathematical modelling and simulation and referring the Graph Net work for my project. I am facing issue saving the model.(Unable to save it.) Can you help me how can I save the model?

This will be of great help to my project.

dbushpw commented 3 years ago

You can save a model in tensorflow at the location checkpoint_path with: model.save_weights(checkpoint_path) And to load it you can use: model.load_weights(checkpoint_path)