Skip to content
Snippets Groups Projects
Commit f8b2afb9 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

Mention where the model will be saved

parent 86d07bc5
Branches
Tags
1 merge request!47Many changes
...@@ -77,6 +77,8 @@ def main(argv=None): ...@@ -77,6 +77,8 @@ def main(argv=None):
exit_ok_exceptions = getattr(config, 'exit_ok_exceptions', exit_ok_exceptions = getattr(config, 'exit_ok_exceptions',
(EarlyStopException,)) (EarlyStopException,))
logger.info("Training a model in %s", estimator.model_dir)
# Train and evaluate # Train and evaluate
try: try:
tf.estimator.train_and_evaluate(estimator, train_spec, eval_spec) tf.estimator.train_and_evaluate(estimator, train_spec, eval_spec)
......
...@@ -73,6 +73,7 @@ def main(argv=None): ...@@ -73,6 +73,7 @@ def main(argv=None):
train_input_fn = config.train_input_fn train_input_fn = config.train_input_fn
# Train # Train
logger.info("Training a model in %s", estimator.model_dir)
estimator.train( estimator.train(
input_fn=train_input_fn, hooks=hooks, steps=steps, max_steps=max_steps) input_fn=train_input_fn, hooks=hooks, steps=steps, max_steps=max_steps)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment