def dump_model(saver, sess, is_best, opt_data):
out_dir = self.helper.checkpoints_saver.get_dir_to_write()
TrainConfigRW(out_dir).save(self.out_config)
save_config(osp.join(out_dir, 'model.config'), self.tf_config)
model_fpath = os.path.join(out_dir, 'model_weights', 'model.ckpt')
saver.save(sess, model_fpath)
self.helper.checkpoints_saver.saved(is_best, opt_data)
def _dump_model_weights(self, out_dir):
save_config(osp.join(out_dir, 'model.config'), self.tf_config)
model_fpath = os.path.join(out_dir, 'model_weights', 'model.ckpt')
self.saver.save(self.sess, model_fpath)