Uses of Class
io.github.kirstenali.deepj.training.Trainer
Packages that use Trainer
Package
Description
Neural networks composed from DeepJ tensors, layers, autograd and backend operations.
-
Uses of Trainer in io.github.kirstenali.deepj.nn
Methods in io.github.kirstenali.deepj.nn that return TrainerModifier and TypeMethodDescriptionstatic TrainerNdSupervisedTraining.trainer(Layer model, LossFunction loss, ParameterOptimizer optimizer, Tensor inputs, Tensor targets, long seed) -
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)