Lightning Utils
PyTorch lightning utilities.
read_csv_logger(experiment_name, log_folder)
Return the loss curves from the csv logs.
Parameters:
| Name | Type | Description | Default |
|---|---|---|---|
experiment_name
|
str
|
Name of the experiment. |
required |
log_folder
|
Path or str
|
Path to the folder containing the csv logs. |
required |
Returns:
| Type | Description |
|---|---|
dict
|
Dictionary containing the loss curves, with keys "train_epoch", "val_epoch", "train_loss" and "val_loss". |