tabensemb.model.AbstractNN.untoggle_optimizer#

method

AbstractNN.untoggle_optimizer(optimizer_idx: int) None#

Resets the state of required gradients that were toggled with toggle_optimizer().

This is only called automatically when automatic optimization is enabled and multiple optimizers are used.

Parameters:

optimizer_idx – The index of the optimizer to untoggle.