core.trainers.base
core.trainers.base
Module for customized trainers
Classes
| Name | Description |
|---|---|
| AxolotlTrainer | Extend the base Trainer for axolotl helpers |
AxolotlTrainer
core.trainers.base.AxolotlTrainer(
*_args,
bench_data_collator=None,
eval_data_collator=None,
dataset_tags=None,
**kwargs,
)Extend the base Trainer for axolotl helpers
Methods
| Name | Description |
|---|---|
| log | Log logs on the various objects watching training, including stored metrics. |
| push_to_hub | Overwrite the push_to_hub method in order to force-add the tags when pushing the |
log
core.trainers.base.AxolotlTrainer.log(logs, start_time=None)Log logs on the various objects watching training, including stored metrics.
Parameters
| Name | Type | Description | Default |
|---|---|---|---|
| logs | dict[str, float] | The values to log. | required |
| start_time | float | None | The start of training. | None |
push_to_hub
core.trainers.base.AxolotlTrainer.push_to_hub(*args, **kwargs)Overwrite the push_to_hub method in order to force-add the tags when pushing the
model on the Hub. Please refer to ~transformers.Trainer.push_to_hub for more details.