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

Fix PR

Browse Source
This commit is contained in:
allegroai 2019-07-24 01:44:09 +03:00
parent 9e166137b6
commit 120de5c24f
2 changed files with 20 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
trains/binding/frameworks/tensorflow_bind.py
View File

@ -154,7 +154,7 @@ class EventTrainsWriter(object):
self.histogram_granularity = histogram_granularity self.histogram_granularity = histogram_granularity
self.histogram_update_freq_multiplier = histogram_update_freq_multiplier self.histogram_update_freq_multiplier = histogram_update_freq_multiplier
self._logger = logger self._logger = logger
self._visualization_mode = 'BGR' self._visualization_mode = 'RGB' # 'BGR'
self._variants = defaultdict(lambda: ()) self._variants = defaultdict(lambda: ())
self._scalar_report_cache = {} self._scalar_report_cache = {}
self._hist_report_cache = {} self._hist_report_cache = {}
@ -582,7 +582,7 @@ class PatchSummaryToEventTransformer(object):
setattr(SummaryToEventTransformer, 'trains', setattr(SummaryToEventTransformer, 'trains',
property(PatchSummaryToEventTransformer.trains_object)) property(PatchSummaryToEventTransformer.trains_object))
except Exception as ex: except Exception as ex:
getLogger(TrainsFrameworkAdapter).warning(str(ex)) getLogger(TrainsFrameworkAdapter).debug(str(ex))
if 'torch' in sys.modules: if 'torch' in sys.modules:
try: try:
@ -596,7 +596,7 @@ class PatchSummaryToEventTransformer(object):
# this is a new version of TensorflowX # this is a new version of TensorflowX
pass pass
except Exception as ex: except Exception as ex:
getLogger(TrainsFrameworkAdapter).warning(str(ex)) getLogger(TrainsFrameworkAdapter).debug(str(ex))
if 'tensorboardX' in sys.modules: if 'tensorboardX' in sys.modules:
try: try:
@ -612,7 +612,7 @@ class PatchSummaryToEventTransformer(object):
# this is a new version of TensorflowX # this is a new version of TensorflowX
pass pass
except Exception as ex: except Exception as ex:
getLogger(TrainsFrameworkAdapter).warning(str(ex)) getLogger(TrainsFrameworkAdapter).debug(str(ex))
if PatchSummaryToEventTransformer.__original_getattributeX is None: if PatchSummaryToEventTransformer.__original_getattributeX is None:
try: try:
@ -626,7 +626,7 @@ class PatchSummaryToEventTransformer(object):
# this is a new version of TensorflowX # this is a new version of TensorflowX
pass pass
except Exception as ex: except Exception as ex:
getLogger(TrainsFrameworkAdapter).warning(str(ex)) getLogger(TrainsFrameworkAdapter).debug(str(ex))
@staticmethod @staticmethod
def _patched_add_eventT(self, *args, **kwargs): def _patched_add_eventT(self, *args, **kwargs):
@ -871,7 +871,7 @@ class PatchTensorFlowEager(object):
except ImportError: except ImportError:
pass pass
except Exception as ex: except Exception as ex:
getLogger(TrainsFrameworkAdapter).warning(str(ex)) getLogger(TrainsFrameworkAdapter).debug(str(ex))
@staticmethod @staticmethod
def _get_event_writer(writer): def _get_event_writer(writer):
@ -1244,17 +1244,17 @@ class PatchTensorflowModelIO(object):
try: try:
# make sure we import the correct version of save # make sure we import the correct version of save
import tensorflow import tensorflow
from tf.saved_model import save from tensorflow.saved_model import save
# actual import # actual import
import tensorflow.saved_model.experimental as saved_model from tensorflow.python.saved_model import save as saved_model
except ImportError: except ImportError:
# noinspection PyBroadException # noinspection PyBroadException
try: try:
# make sure we import the correct version of save # make sure we import the correct version of save
import tensorflow import tensorflow
from tensorflow.saved_model import save from tensorflow.saved_model.experimental import save
# actual import # actual import
import tensorflow.saved_mode as saved_model import tensorflow.saved_model.experimental as saved_model
except ImportError: except ImportError:
saved_model = None saved_model = None
except Exception: except Exception:

13
trains/utilities/seed.py
View File

@ -66,11 +66,18 @@ def make_deterministic(seed=1337, cudnn_deterministic=False):
try: try:
tf.compat.v1.set_random_seed(seed) tf.compat.v1.set_random_seed(seed)
except Exception: except Exception:
pass try:
tf.set_random_seed(seed)
except Exception:
pass
try: try:
tf.compat.v1.set_random_seed(seed) tf.compat.v1.random.set_random_seed(seed)
except Exception: except Exception:
pass try:
tf.random.set_random_seed(seed)
except Exception:
pass
make_deterministic() make_deterministic()