TuningMixin

class liesel.goose.kernel.TuningMixin

Bases: Generic[TKernelState, TTuningInfo]

An abstract mixin defining two tuning methods after a slow and a fast adaptation epoch.

Methods

tune(prng_key, kernel_state, model_state, ...)

The method can perform automatic tuning of the kernel and is called after each adaptation epoch.