RWKernel.end_epoch()

RWKernel.end_epoch()#

RWKernel.end_epoch(prng_key, kernel_state, model_state, epoch)[source]#

Sets the step size as found by the dual averaging algorithm.

Return type:

RWKernelState