NaNMonitor

class composer.callbacks.NaNMonitor(*args, **kwargs)

Catches NaNs in the loss and raises an error if one is found.

after_loss(state, logger)

Check if loss is NaN and raise an error if so.