Method _training_step

utils/commons/base_task.py:96–103

:param sample: :param batch_idx: :return: total loss: torch.Tensor, loss_log: dict

(self, sample, batch_idx, optimizer_idx)

        return loss_outputs

    def _training_step(self, sample, batch_idx, optimizer_idx):
        """

        :param sample:
        :param batch_idx:
        :return: total loss: torch.Tensor, loss_log: dict
        """
        raise NotImplementedError

    def training_step(self, sample, batch_idx, optimizer_idx=-1):
        """