sup3r.models.abstract.TensorboardMixIn

class TensorboardMixIn

Bases: object

MixIn class for tensorboard logging and profiling.

Methods

dict_to_tensorboard(entry)	Write data to tensorboard log file.
profile_to_tensorboard(name)	Write profile data to tensorboard log file.

Attributes

total_batches

Record of total number of batches for logging.

property total_batches

Record of total number of batches for logging.

dict_to_tensorboard(entry)

Write data to tensorboard log file. This is usually a loss_details dictionary.

Parameters:

entry (dict) – Dictionary of values to write to tensorboard log file

profile_to_tensorboard(name)

Write profile data to tensorboard log file.

Parameters:

name (str) – Tag name to use for profile info