From b7688f251df7ce3c289477bdc77d6a232934b3ea Mon Sep 17 00:00:00 2001 From: Thomas Deegan Date: Tue, 27 Nov 2018 15:49:23 -0800 Subject: [PATCH] Make var loading logging messages debug from info You may load a large amount of variables from a checkpoint and printing them all to "INFO" is not very useful as it spams the log. I propose downgrading these to debug so that they only show up when you are trying to "debug" your loading. --- tensorflow/python/training/warm_starting_util.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/python/training/warm_starting_util.py b/tensorflow/python/training/warm_starting_util.py index 19dc04e8fb2..ef438f9c4de 100644 --- a/tensorflow/python/training/warm_starting_util.py +++ b/tensorflow/python/training/warm_starting_util.py @@ -434,7 +434,7 @@ def warm_start(ckpt_to_initialize_from, vocab_info = var_name_to_vocab_info.get(var_name) if vocab_info: vocab_info_used.add(var_name) - logging.info( + logging.debug( "Warm-starting variable: {}; current_vocab: {} current_vocab_size: {}" " prev_vocab: {} prev_vocab_size: {} current_oov: {} prev_tensor: {}" " initializer: {}".format( @@ -462,7 +462,7 @@ def warm_start(ckpt_to_initialize_from, # For the special value of vars_to_warm_start = None, # we only warm-start variables with explicitly specified vocabularies. if vars_to_warm_start: - logging.info("Warm-starting variable: {}; prev_var_name: {}".format( + logging.debug("Warm-starting variable: {}; prev_var_name: {}".format( var_name, prev_var_name or "Unchanged")) # Because we use a default empty list in grouped_variables, single # unpartitioned variables will be lists here, which we rectify in order