graphorge.gnn_base_model.train.training.read_lr_history_from_file

read_lr_history_from_file(loss_record_path)[source]

Read training learning rate history from loss history record file.

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

Parameters:

loss_record_path (str) – Loss history record file path.

Returns:

  • lr_scheduler_type ({‘steplr’, ‘explr’, ‘linlr’}) – Type of learning rate scheduler:

    ’steplr’ : Step-based decay (torch.optim.lr_scheduler.SetpLR)

    ’explr’ : Exponential decay (torch.optim.lr_scheduler.ExponentialLR)

    ’linlr’ : Linear decay (torch.optim.lr_scheduler.LinearLR)

  • lr_history_epochs (list[float]) – Training process learning rate history (per epoch).