Skip to content

GitLab

Commit e387ed65 authored by Yuexin Wu's avatar Yuexin Wu Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 407125012
parent d7940205
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/core/train_utils.py
View file @ e387ed65
...@@ -311,10 +311,10 @@ def serialize_config(params: config_definitions.ExperimentConfig, ...@@ -311,10 +311,10 @@ def serialize_config(params: config_definitions.ExperimentConfig,
hyperparams.save_params_dict_to_yaml(params, params_save_path) hyperparams.save_params_dict_to_yaml(params, params_save_path)
def save_gin_config(filename_surfix: str, model_dir: str): def save_gin_config(filename_suffix: str, model_dir: str):
"""Serializes and saves the experiment config.""" """Serializes and saves the experiment config."""
gin_save_path = os.path.join( gin_save_path = os.path.join(
model_dir, 'operative_config.{}.gin'.format(filename_surfix)) model_dir, 'operative_config.{}.gin'.format(filename_suffix))
logging.info('Saving gin configurations to %s', gin_save_path) logging.info('Saving gin configurations to %s', gin_save_path)
tf.io.gfile.makedirs(model_dir) tf.io.gfile.makedirs(model_dir)
with tf.io.gfile.GFile(gin_save_path, 'w') as f: with tf.io.gfile.GFile(gin_save_path, 'w') as f:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment