OptimNaNDebugInfo.reproduce_loss()

OptimNaNDebugInfo.reproduce_loss()#

OptimNaNDebugInfo.reproduce_loss(engine)[source]#

Reproduces the captured pre-update loss.

Optimizer-associated events replay that optimizer’s step from the saved state and return its scalar loss. Events captured while no optimizer was active re-evaluate the explicit batched training loss directly.

Return type:

Array