Fix Tensorboard RGB channel order

This commit is contained in:
allegroai
2019-07-14 19:04:01 +03:00
parent 7199dc5d55
commit 44aaa39ccf
2 changed files with 8 additions and 5 deletions

View File

@@ -93,7 +93,7 @@ class EventTrainsWriter(object):
self.histogram_granularity = histogram_granularity
self.histogram_update_freq_multiplier = histogram_update_freq_multiplier
self._logger = logger
self._visualization_mode = 'BGR'
self._visualization_mode = 'RGB' # 'BGR'
self._variants = defaultdict(lambda: ())
self._scalar_report_cache = {}
self._hist_report_cache = {}