MultiPassOptimizer

An optimizer that performs multiple passes over training data, updating the model parameters multiple times per epoch.

Functions

Creates training batches out of the given cases.

abstract fun passAgain(): Boolean

Returns true if the optimizer should perform another pass over the training data.

abstract fun update(step: Int, epoch: Int, layer: Layer, weightGradients: Tensor, biasGradients: Tensor)

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