Skip to content

Hf example

HFExample #

Bases: LightningModule

Example of a lightning module used to train a huggingface model.

configure_optimizers #

configure_optimizers()

Prepare optimizer and schedule (linear warmup and decay)