xlstm_jax.start_training

Functions

start_training(cfg)

Starts training from scratch.

Module Contents

xlstm_jax.start_training.start_training(cfg)

Starts training from scratch.

Decorated with error catching decorator.

Parameters:

cfg (omegaconf.DictConfig) – The full configuration.

Returns:

The final metrics of the resumed training.

Return type:

dict[str, Any]