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: