Skip to content

GitLab

Commit bd217c64 authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Change logging about the best checkpoint export 

PiperOrigin-RevId: 345054031
parent 29120423
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
official/core/train_lib.py
View file @ bd217c64
...@@ -40,12 +40,13 @@ def maybe_create_best_ckpt_exporter(params: config_definitions.ExperimentConfig, ...@@ -40,12 +40,13 @@ def maybe_create_best_ckpt_exporter(params: config_definitions.ExperimentConfig,
best_ckpt_dir = os.path.join(data_dir, export_subdir) best_ckpt_dir = os.path.join(data_dir, export_subdir)
best_ckpt_exporter = BestCheckpointExporter( best_ckpt_exporter = BestCheckpointExporter(
best_ckpt_dir, metric_name, metric_comp) best_ckpt_dir, metric_name, metric_comp)
else:
best_ckpt_exporter = None
logging.info( logging.info(
'Not exporting the best checkpoint. ' 'Created the best checkpoint exporter. '
'data_dir: %s, export_subdir: %s, metric_name: %s', data_dir, 'data_dir: %s, export_subdir: %s, metric_name: %s', data_dir,
export_subdir, metric_name) export_subdir, metric_name)
else:
best_ckpt_exporter = None
return best_ckpt_exporter return best_ckpt_exporter
......
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