Skip to content

Csv Logger

Source

PyTorch lightning utilities.

Series dataclass

An easy to plot series extracting from training log.

epoch instance-attribute

X-axis values representing training epochs.

value instance-attribute

Y-axis values representing the quantity recorded over the epoch.

TrainingReport dataclass

Series extracted from training logs.

learning_rate instance-attribute

Learning rate.

metrics instance-attribute

Metrics, can be empty.

train_loss instance-attribute

Training loss.

val_loss instance-attribute

Validation loss.

read_csv_logger(log_folder, experiment_name, version=None)

Return plottable training curves from Lightning CSV logs.

Parameters:

  • log_folder (Path or str) –

    Path to the folder containing the csv logs.

  • experiment_name (str) –

    Name of the experiment.

  • version (int or None, default: = None ) –

    Version number to load, if None then the latest version is loaded.

Returns:

  • TrainingReport –

    Dataclass containing train and validation loss, learning rate, and any discovered validation metrics.

Raises:

  • ValueError –

    If version is specified but not found.