trainers.Difftre.restore

Difftre.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.