Merge pull request #468 from NifTK/fixes-issue-454

fixes #454
NifTK · Dec 18, 2019 · 9b12da4 · 9b12da4
2 parents a9b30a6 + 32641e5
commit 9b12da4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/niftynet/engine/handler_performance.py b/niftynet/engine/handler_performance.py
@@ -36,6 +36,6 @@ def update_performance_history(self, _sender, **msg):
  else:
  _sender.performance_history = \
  _sender.performance_history[1:] + [current_loss]
- except AttributeError:
+ except (AttributeError, KeyError):
  tf.logging.warning("does not contain any performance field "
  "called total loss.")