From 332e9e2f63738b1c04b91d7878b41d2f1351d4c7 Mon Sep 17 00:00:00 2001 From: allegroai <> Date: Sun, 22 Mar 2020 18:17:16 +0200 Subject: [PATCH] Fix Tensorflow direct V2.1 multiple FileWriters --- trains/binding/frameworks/tensorflow_bind.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/trains/binding/frameworks/tensorflow_bind.py b/trains/binding/frameworks/tensorflow_bind.py index 0afc93ed..fbc7b159 100644 --- a/trains/binding/frameworks/tensorflow_bind.py +++ b/trains/binding/frameworks/tensorflow_bind.py @@ -1006,9 +1006,18 @@ class PatchTensorFlowEager(object): try: from tensorflow.python.eager import context logdir = context.context().summary_writer._init_op_fn.keywords.get('logdir') - logdir = logdir.numpy().decode() + except: + try: + from tensorflow.python.ops.summary_ops_v2 import _summary_state + logdir = _summary_state.writer._init_op_fn.keywords.get('logdir') + except: + logdir = None + try: + if logdir is not None: + logdir = logdir.numpy().decode() except: logdir = None + PatchTensorFlowEager.__trains_event_writer[id(writer)] = EventTrainsWriter( logger=PatchTensorFlowEager.__main_task.get_logger(), logdir=logdir, **PatchTensorFlowEager.defaults_dict)