CompositeLogger

CompositeLogger#

class braintools.trainer.CompositeLogger(loggers)#

Combine multiple loggers.

Logs to all provided loggers simultaneously.

Parameters:

loggers (List[Logger]) – List of loggers to use.

Examples

>>> logger = CompositeLogger([
...     TensorBoardLogger('logs/'),
...     CSVLogger('logs/'),
... ])
>>> trainer = Trainer(logger=logger)
finalize(status='success')[source]#

Finalize logging (called at end of training).

Parameters:

status (str) – Final status (‘success’, ‘failed’, ‘interrupted’).

log_artifact(local_path, artifact_path=None)[source]#

Log an artifact (file).

Parameters:

  • local_path (str) – Path to the local file.

  • artifact_path (str | None) – Path in artifact storage.

log_graph(model, input_array=None)[source]#

Log model graph/architecture.

Parameters:

  • model (Any) – The model to log.

  • input_array (Any) – Sample input for tracing.

log_hyperparams(params)[source]#

Log hyperparameters.

Parameters:

params (Dict[str, Any]) – Dictionary of hyperparameter names to values.

log_image(key, images, step=None)[source]#

Log images.

Parameters:

  • key (str) – Image key/tag.

  • images (Any) – Images to log (numpy array or list).

  • step (int | None) – Global step number.

log_metrics(metrics, step=None)[source]#

Log metrics.

Parameters:

  • metrics (Dict[str, float]) – Dictionary of metric names to values.

  • step (int | None) – Global step number.

log_text(key, text, step=None)[source]#

Log text.

Parameters:

  • key (str) – Text key/tag.

  • text (str) – Text to log.

  • step (int | None) – Global step number.

property name: str#

Experiment name.

save()[source]#

Save/flush any buffered data.

property version: str | None#

Experiment version/run ID.