Uses of Interface
io.github.kirstenali.deepj.optimisers.ParameterOptimizer
Packages that use ParameterOptimizer
Package
Description
Neural networks composed from DeepJ tensors, layers, autograd and backend operations.
-
Uses of ParameterOptimizer in io.github.kirstenali.deepj.nn
Methods in io.github.kirstenali.deepj.nn with parameters of type ParameterOptimizerModifier and TypeMethodDescriptionstatic TrainerNdSupervisedTraining.trainer(Layer model, LossFunction loss, ParameterOptimizer optimizer, Tensor inputs, Tensor targets, long seed) -
Uses of ParameterOptimizer in io.github.kirstenali.deepj.optimisers
Classes in io.github.kirstenali.deepj.optimisers that implement ParameterOptimizer -
Uses of ParameterOptimizer in io.github.kirstenali.deepj.training
Methods in io.github.kirstenali.deepj.training with parameters of type ParameterOptimizerModifier and TypeMethodDescriptionstatic TrainerCausalLMTraining.trainer(CausalLM model, BatchSource dataset, ParameterOptimizer optimizer) static TrainerSupervisedTraining.trainer(Layer model, LossFunction lossFn, ParameterOptimizer opt, Tensor xAll, Tensor yAll, long seed)