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(CausalLM model, BatchSource dataset, float lr)
static Trainer	trainer(CausalLM model, BatchSource dataset, ParameterOptimizer optimizer)

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(CausalLM model,
 BatchSource dataset,
 float lr)

trainer

public static Trainer trainer(CausalLM model,
 BatchSource dataset,
 ParameterOptimizer optimizer)