feature-training

provider(?)

feature-training/org.pointyware.disco.training.entities.optimizers/WarmRestartExponentialLearningRate

WarmRestartExponentialLearningRate

data class WarmRestartExponentialLearningRate(val initialLearningRate: Float, val decayMax: Float = 100.0f, val decayPeriod: Int = 10000) : LearningRateSchedule

This learning rate schedule begins at the initialLearningRate and decays exponentially until it reaches initialLearningRate / decayMax over decayPeriod epochs.

Constructors

WarmRestartExponentialLearningRate

constructor(initialLearningRate: Float, decayMax: Float = 100.0f, decayPeriod: Int = 10000)

Properties

val decayMax: Float = 100.0f

val decayPeriod: Int = 10000

initialLearningRate

val initialLearningRate: Float

Functions

open override fun learningRate(epoch: Int): Float