Commit 057bcf3d authored by Felix Bragman's avatar Felix Bragman

output all trained variables as histograms to tf summaries

parent 1e08b679
Pipeline #12603 failed with stages
in 9 seconds
......@@ -182,6 +182,11 @@ class OutputsCollector(object):
This function should be called in
`ApplicationDriver.create_graph` function
"""
all_vars = tf.get_collection(tf.GraphKeys.TRAINABLE_VARIABLES)
for var in all_vars:
tf.summary.histogram(var.name, var)
self._average_variables_over_devices(self.console_vars, False)
self._average_variables_over_devices(self.output_vars, False)
self._average_variables_over_devices(self.summary_vars, True)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment