update

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.

epoch

The current epoch.

layer

The layer whose parameters are being updated.

weightGradients

The gradients of the weights computed during the backward pass.

biasGradients

The gradients of the biases computed during the backward pass.