Skip to content
Snippets Groups Projects
Commit cde558b2 authored by Saeed SARFJOO's avatar Saeed SARFJOO
Browse files

remove checking log_parameters is None

parent 3bab7d65
No related branches found
No related tags found
1 merge request!61add sort_by_accuracy and max_wait_intervals to eval.py
Pipeline #
...@@ -165,8 +165,7 @@ def append_evaluated_file(path, evaluations): ...@@ -165,8 +165,7 @@ def append_evaluated_file(path, evaluations):
def eval(estimator, eval_input_fn, hooks, run_once, eval_interval_secs, name, def eval(estimator, eval_input_fn, hooks, run_once, eval_interval_secs, name,
keep_n_best_models, sort_by_accuracy, max_wait_intervals, **kwargs): keep_n_best_models, sort_by_accuracy, max_wait_intervals, **kwargs):
"""Evaluates networks using Tensorflow estimators.""" """Evaluates networks using Tensorflow estimators."""
if not click.get_current_context(True) is None: log_parameters(logger)
log_parameters(logger)
real_name = 'eval_' + name if name else 'eval' real_name = 'eval_' + name if name else 'eval'
eval_dir = os.path.join(estimator.model_dir, real_name) eval_dir = os.path.join(estimator.model_dir, real_name)
......
...@@ -53,8 +53,7 @@ logger = logging.getLogger(__name__) ...@@ -53,8 +53,7 @@ logger = logging.getLogger(__name__)
@verbosity_option(cls=ResourceOption) @verbosity_option(cls=ResourceOption)
def train(estimator, train_input_fn, hooks, steps, max_steps, **kwargs): def train(estimator, train_input_fn, hooks, steps, max_steps, **kwargs):
"""Trains networks using Tensorflow estimators.""" """Trains networks using Tensorflow estimators."""
if not click.get_current_context(True) is None: log_parameters(logger)
log_parameters(logger)
# Train # Train
logger.info("Training a model in %s", estimator.model_dir) logger.info("Training a model in %s", estimator.model_dir)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment