From 2682754aaf971b5d13fc46dca66619cf3290274f Mon Sep 17 00:00:00 2001 From: allegroai <> Date: Sat, 6 Mar 2021 00:46:22 +0200 Subject: [PATCH] Fix TF bind keyboard interrupt --- clearml/binding/frameworks/tensorflow_bind.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/clearml/binding/frameworks/tensorflow_bind.py b/clearml/binding/frameworks/tensorflow_bind.py index 573e0778..fd113d4c 100644 --- a/clearml/binding/frameworks/tensorflow_bind.py +++ b/clearml/binding/frameworks/tensorflow_bind.py @@ -375,6 +375,8 @@ class EventTrainsWriter(object): val = val[:, :, [2, 1, 0]] else: val = val[:, :, [0, 1, 2]] + except KeyboardInterrupt: + raise except Exception as e: logger = LoggerRoot.get_base_logger(TensorflowBinding) logger.warning('Failed decoding debug image [%s, %s, %s]' % (width, height, color_channels))