def _save_config(self):
"""Save parameters to config files if model_dir is defined."""
model_dir = self._flags.model_dir
if model_dir is not None:
if not tf.gfile.Exists(model_dir):
tf.gfile.MakeDirs(model_dir)
params_io.save_hparams_to_yaml(self._model_config,
model_dir + '/params.yaml')
def save_config(config, model_dir):
"""Save parameters to config files."""
if not tf.gfile.Exists(model_dir):
tf.gfile.MakeDirs(model_dir)
params_io.save_hparams_to_yaml(config, model_dir + '/params.yaml')