api/org.diffkt.model/Optimizer

Optimizer

abstract class Optimizer<T : Trainable<T>>

Functions

tensorTrainingStep

abstract fun tensorTrainingStep(tensor: DTensor, gradient: DTensor): DTensor

Train an element of the model. For the model itself, use train.

train

fun train(component: T, tangent: Trainable.Tangent): T

Train a trainable using the given tangent.

Inheritors

AdamOptimizer

FixedLearningRateOptimizer

RMSpropOptimizer

SGDOptimizer