(self, title: str = None)
Construct summary ops to include histograms of all trainable parameters in TensorBoard.
Construct summary ops to include histograms of all trainable parameters in TensorBoard.
Construct summary ops to include histograms of all trainable parameters in TensorBoard .
def setup_weight_histograms(self, title: str = None) -> None:
        """Construct summary ops to include histograms of all trainable parameters in TensorBoard."""
        if title is None:
            title = self.name

        with tf.name_scope(None), tf.device(None), tf.control_dependencies(None):
            for local_name, var in self.trainables.items():
                if "/" in local_name:
                    p = local_name.split("/")
                    name = title + "_" + p[-1] + "/" + "_".join(p[:-1])
                else:
                    name = title + "_toplevel/" + local_name

                tf.summary.histogram(name, var)
def setup_weight_histograms ( self , title : str = None ) -> None : if title is None : title = self . name with tf . name_scope ( None ) , tf . device ( None ) , tf . control_dependencies ( None ) : for local_name , var in self . trainables . items ( ) : if "/" in local_name : p = local_name . split ( "/" ) name = title + "_" + p [ - 1 ] + "/" + "_" . join ( p [ : - 1 ] ) else : name = title + "_toplevel/" + local_name tf . summary . histogram ( name , var )