api/org.diffkt.model/FixedLearningRateOptimizer

FixedLearningRateOptimizer

class FixedLearningRateOptimizer<T : Model<T>>(alpha: DScalar) : Optimizer<T>

simple optimizer that just uses a fixed learning rate.

Constructors

FixedLearningRateOptimizer

fun FixedLearningRateOptimizer(alpha: Float)

FixedLearningRateOptimizer

fun FixedLearningRateOptimizer(alpha: DScalar)

Functions

tensorTrainingStep

open override 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.

Properties

alpha

val alpha: DScalar