ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-01-31 17:17:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix TF bind keyboard interrupt

Browse Source
This commit is contained in:
allegroai 2021-03-06 00:46:22 +02:00
parent 912f6f5ba2
commit 2682754aaf
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
clearml/binding/frameworks/tensorflow_bind.py
View File

@ -375,6 +375,8 @@ class EventTrainsWriter(object):
val = val[:, :, [2, 1, 0]] val = val[:, :, [2, 1, 0]]
else: else:
val = val[:, :, [0, 1, 2]] val = val[:, :, [0, 1, 2]]
except KeyboardInterrupt:
raise
except Exception as e: except Exception as e:
logger = LoggerRoot.get_base_logger(TensorflowBinding) logger = LoggerRoot.get_base_logger(TensorflowBinding)
logger.warning('Failed decoding debug image [%s, %s, %s]' % (width, height, color_channels)) logger.warning('Failed decoding debug image [%s, %s, %s]' % (width, height, color_channels))