trainers.ForceMatching.restore

ForceMatching.restore(checkpoint)

Restores the trainer from a checkpoint.

Parameters:

checkpoint – Checkpoint to restore from. Can be a path to a file or a dictionary containing the trainer state.