xlstm_jax.resume_training

Functions

resume_training(cfg)

Resumes training from a checkpoint.

Module Contents

xlstm_jax.resume_training.resume_training(cfg)

Resumes training from a checkpoint.

Parameters:

cfg (omegaconf.DictConfig) – The full configuration.

Returns:

The final metrics of the resumed training.

Return type:

dict[str, Any]