diff options
author | 2017-01-11 17:10:10 -0800 | |
---|---|---|
committer | 2017-01-11 17:24:51 -0800 | |
commit | bcdecca6c913900100bb123f3f1cb343bc0c9083 (patch) | |
tree | 099ed77a664e9db1f35a7a76adcf62506a702746 | |
parent | 112c2087ab39b8cfb1610720b193632ed5cd3211 (diff) |
Estimator fit method returns self even if max_steps has been reached.
Change: 144269545
-rw-r--r-- | tensorflow/contrib/learn/python/learn/estimators/estimator.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/learn/python/learn/estimators/estimator.py b/tensorflow/contrib/learn/python/learn/estimators/estimator.py index 467d31c331..381fbffac3 100644 --- a/tensorflow/contrib/learn/python/learn/estimators/estimator.py +++ b/tensorflow/contrib/learn/python/learn/estimators/estimator.py @@ -414,7 +414,7 @@ class BaseEstimator( start_step = load_variable(self._model_dir, ops.GraphKeys.GLOBAL_STEP) if max_steps <= start_step: logging.info('Skipping training since max_steps has already saved.') - return None + return self except: # pylint: disable=bare-except pass |