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

RmsProp

class RmsProp : SinglePassOptimizer

Implemented per lecture slides: https://www.cs.toronto.edu/~tijmen/csc321/slides/lecture_slides_lec6.pdf

Constructors

Link copied to clipboard
constructor()

Functions

Link copied to clipboard
open override fun batch(cases: List<Exercise>): List<List<Exercise>>

Creates training batches out of the given cases.

Link copied to clipboard
open override fun update(epoch: Int, layer: Layer, weightGradients: Tensor, biasGradients: Tensor)

Updates the parameters of the model based on the outputs computed during the forward pass.