I am currently using the TensorFlow source to save and restore the trained NN model weights:
# Save the weights model.save_weights(\'./checkpoints/my_checkpoi