gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 11:10:20 +01:00
Code Issues Projects Releases Wiki Activity

fix for tensorboard visualization slowing execution even when it is off

Browse Source 
apparently tensorflow still collect summary data even when no summary FileWriter is defined.
This commit is contained in:
Gal Leibovich
2018-02-17 12:42:59 +02:00
parent 72d34f4063
commit 16c5032735
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
architectures/tensorflow_components/architecture.py
View File

@@ -90,7 +90,8 @@ class TensorFlowArchitecture(Architecture):
for idx, var in enumerate(self.trainable_weights):
placeholder = tf.placeholder(tf.float32, shape=var.get_shape(), name=str(idx) + '_holder')
self.weights_placeholders.append(placeholder)
variable_summaries(var)
if self.tp.visualization.tensorboard:
variable_summaries(var)
self.update_weights_from_list = [weights.assign(holder) for holder, weights in
zip(self.weights_placeholders, self.trainable_weights)]