Skip to content

Csv Logger

Source

PyTorch lightning utilities.

read_csv_logger(experiment_name, log_folder)

Return the loss curves from the csv logs.

Parameters:

  • experiment_name (str) –

    Name of the experiment.

  • log_folder (Path or str) –

    Path to the folder containing the csv logs.

Returns:

  • dict –

    Dictionary containing the loss curves, with keys "train_epoch", "val_epoch", "train_loss" and "val_loss".