graphorge.gnn_base_model.train.training.load_lr_history

load_lr_history(model, epoch=None)[source]

Load training process learning rate history record.

Loss history record file is stored in model_directory under the name loss_history_record.pkl.

Parameters:

  • model (torch.nn.Module) – Model.

  • epoch (int, default=None) – Training epoch to which loss history is loaded (included), with the first training epoch being 0. If None, then loads the full loss history.

Returns:

lr_history_epochs – Training process learning rate history (per epoch).

Return type:

list[float]