def save_config(params, model_dir):
if model_dir:
params = filter_unused_blocks(params)
if not tf.gfile.Exists(model_dir):
tf.gfile.MakeDirs(model_dir)
params_dict.save_params_dict_to_yaml(
params, os.path.join(model_dir, 'params.yaml'))
def _save_params(self):
"""Save parameters to config files if model_dir is defined."""
model_dir = self._model_dir
if model_dir is not None:
if not tf.gfile.Exists(model_dir):
tf.gfile.MakeDirs(model_dir)
params_dict.save_params_dict_to_yaml(self._params,
model_dir + '/params.yaml')
def _save_config(self, model_dir):
"""Save parameters to config files if model_dir is defined."""
logging.info('Save config to model_dir %s.', model_dir)
if model_dir:
if not tf.io.gfile.exists(model_dir):
tf.io.gfile.makedirs(model_dir)
self._params.lock()
params_dict.save_params_dict_to_yaml(self._params,
model_dir + '/params.yaml')
else:
logging.warning('model_dir is empty, so skip the save config.')
def save_config(params, model_dir):
if model_dir:
if not tf.gfile.Exists(model_dir):
tf.gfile.MakeDirs(model_dir)
params_dict.save_params_dict_to_yaml(params, model_dir + '/param.yaml')
