35bcd7ada7488e6a2df941baeaf5c6315ceda207,inferno/trainers/callbacks/essentials.py,NaNDetector,end_of_training_iteration,#NaNDetector#,12

Before Change

        training_loss = self.trainer.get_state("training_loss")
        // Extract scalar
        if tu.is_tensor(training_loss):
            training_loss = training_loss.float()[0]
        if not np.isfinite(training_loss):
            raise RuntimeError("Loss is not finite (loss={})!".format(training_loss))

After Change

        training_loss = self.trainer.get_state("training_loss")
        // Extract scalar
        if tu.is_tensor(training_loss):
            training_loss = tu.unwrap(training_loss, extract_item=True)
        if not np.isfinite(training_loss):
            raise RuntimeError("Loss is not finite (loss={})!".format(training_loss))

Instances