fix for tensorboard visualization slowing execution even when it is off
apparently tensorflow still collect summary data even when no summary FileWriter is defined.
This commit is contained in:
1 parent
72d34f4063
commit
16c5032735
1 file changed
+2
-1
@@ -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)]
|
||||
|
||||
Reference in new issue
Block a user