Class CausalLMTraining
java.lang.Object
io.github.kirstenali.deepj.training.CausalLMTraining
Wiring helpers for causal language model training.
-
Method Summary
Modifier and TypeMethodDescriptionstatic Trainertrainer(GPTModel model, TextDataset dataset, double lr)
-
Method Details
-
trainer
-