Uses of Class
io.github.kirstenali.deepj.training.Trainer
Packages that use Trainer
-
Uses of Trainer in io.github.kirstenali.deepj.training
Methods in io.github.kirstenali.deepj.training that return TrainerModifier and TypeMethodDescriptionstatic TrainerCausalLMTraining.trainer(CausalLM model, BatchSource dataset, float lr) static TrainerCausalLMTraining.trainer(CausalLM model, BatchSource dataset, ParameterOptimizer optimizer) static TrainerSupervisedTraining.trainer(Layer model, LossFunction lossFn, ParameterOptimizer opt, Tensor xAll, Tensor yAll, long seed)