ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-07 13:23:40 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix some plots and debug samples are missing for clearml examples

Browse Source
This commit is contained in:
allegroai 2023-02-28 17:10:48 +02:00
parent 5d027a335f
commit 22715cda19
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -388,6 +388,7 @@ class EventTrainsWriter(object):
imdata = base64.b64decode(img_str) imdata = base64.b64decode(img_str)
output = BytesIO(imdata) output = BytesIO(imdata)
im = Image.open(output) im = Image.open(output)
image = np.asarray(im)
# if this is a GIF store as is # if this is a GIF store as is
if getattr(im, 'is_animated', None): if getattr(im, 'is_animated', None):
output.close() output.close()
@ -398,7 +399,6 @@ class EventTrainsWriter(object):
with open(fd, "wb") as f: with open(fd, "wb") as f:
f.write(imdata) f.write(imdata)
return temp_file return temp_file
image = np.asarray(im)
output.close() output.close()
if height is not None and height > 0 and width is not None and width > 0: if height is not None and height > 0 and width is not None and width > 0:
val = image.reshape((height, width, -1)).astype(np.uint8) val = image.reshape((height, width, -1)).astype(np.uint8)
@ -731,8 +731,8 @@ class EventTrainsWriter(object):
step -= EventTrainsWriter._current_task.get_initial_iteration() step -= EventTrainsWriter._current_task.get_initial_iteration()
# there can be a few metrics getting reported again, so the step can be negative # there can be a few metrics getting reported again, so the step can be negative
# for the first few reports # for the first few reports
if step <= 0: if step < 0:
return step = 0
self._max_step = max(self._max_step, step) self._max_step = max(self._max_step, step)
if value_dicts is None: if value_dicts is None:
LoggerRoot.get_base_logger(TensorflowBinding).debug("Summary arrived without 'value'") LoggerRoot.get_base_logger(TensorflowBinding).debug("Summary arrived without 'value'")