Add model.save to end of all baseline scripts
dustinvtran opened this issue · comments
Some scripts don't produce a SavedModel at the end. We should double check this for consistency.
mentioned by @JeremyNixon
I'd be interested in doing a quick check. Just to clarify, model.save refers to this snippet, right?
final_save_name = os.path.join(FLAGS.output_dir, 'model')
model.save(final_save_name)
logging.info('Saved model to %s', final_save_name)