Package io.github.kirstenali.deepj.training

Class CausalLMTraining

java.lang.Object

io.github.kirstenali.deepj.training.CausalLMTraining

public final class CausalLMTraining
extends Object

Wiring helpers for causal language model training.

Method Summary

Modifier and Type	Method	Description
static Trainer	trainer(GPTModel model, TextDataset dataset, double lr)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Details

trainer

public static Trainer trainer(GPTModel model,
 TextDataset dataset,
 double lr)