tabensemb.model.AbstractNN.on_after_backward

method

AbstractNN.on_after_backward() → None

Called after loss.backward() and before optimizers are stepped.

Note

If using native AMP, the gradients will not be unscaled at this point. Use the on_before_optimizer_step if you need the unscaled gradients.