api/org.diffkt.model/AdamOptimizer

AdamOptimizer

class AdamOptimizer<T : Model<T>> : Optimizer<T>
Content copied to clipboard

THIS IS A PLACEHOLDER!

Constructors

AdamOptimizer
Link copied to clipboard
fun AdamOptimizer()
Content copied to clipboard

Functions

tensorTrainingStep
Link copied to clipboard
open override fun tensorTrainingStep(tensor: DTensor, gradient: DTensor): DTensor
Content copied to clipboard

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

train
Link copied to clipboard
fun train(component: T, tangent: Trainable.Tangent): T
Content copied to clipboard

Train a trainable using the given tangent.