in sample_workloads/lit-gpt-demo/openwebtext.py [0:0]
def train(
fabric: L.Fabric,
devices: int,
state: dict,
train_dataloader: DataLoader,
val_dataloader: DataLoader,
io_args: IOArgs,
train_args: TrainArgs,
eval_args: EvalArgs,