nvalchemi.training.BatchValidationCallback

class nvalchemi.training.BatchValidationCallback(*args, **kwargs)

Protocol for an optional per-batch validation callback.

A user-supplied object implementing this protocol is invoked once per validation batch inside ValidationLoop.execute(), immediately after predictions and the per-batch loss are computed. It is the extension point for streaming per-batch outputs (e.g. predictions or diagnostics) to a custom logging or storage system.

Summary-level logging does not require this callback: register a hook on AFTER_VALIDATION and read the validation summary from ctx.validation.

Notes

No concrete implementation is provided. Users supply their own.