sup3r.models.abstract.TensorboardMixIn

class TensorboardMixIn

Bases: object

MixIn class for tensorboard logging and profiling.

Note: To monitor model training with tensorboard run ssh -N -f -L localhost:6006:localhost:6006 <user>@<host> on your local machine and then tensorboard --logdir <log_dir> on the remote machine

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