From 79e136bd03ffd957b156677cad72894002ad8634 Mon Sep 17 00:00:00 2001 From: LielinJiang Date: Thu, 7 May 2020 08:20:25 +0000 Subject: [PATCH] fix eval final steps bug --- hapi/callbacks.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/hapi/callbacks.py b/hapi/callbacks.py index f7bb878..c13fd43 100644 --- a/hapi/callbacks.py +++ b/hapi/callbacks.py @@ -311,7 +311,8 @@ class ProgBarLogger(Callback): def on_eval_end(self, logs=None): logs = logs or {} - if self._is_print() and (self.steps is not None): + + if self._is_print() and (self.eval_steps is not None): self._updates(logs, 'eval') print('Eval samples: %d' % (self.evaled_samples)) -- GitLab