core.trainers.mamba

core.trainers.mamba

Module for mamba trainer

Classes

Name	Description
AxolotlMambaTrainer	Mamba specific trainer to handle loss calculation

AxolotlMambaTrainer

core.trainers.mamba.AxolotlMambaTrainer(
    *_args,
    bench_data_collator=None,
    eval_data_collator=None,
    dataset_tags=None,
    **kwargs,
)

Mamba specific trainer to handle loss calculation