Method train_step

cdslib/core/script/base_train.py:1095–1105

One training step. Return a dictionary that will be passed to logging.

(
            self,
            epoch: int,
            bidx: int,
            batch: T.Any,
    )

    @customer
    @abstractmethod
    def train_step(
            self,
            epoch: int,
            bidx: int,
            batch: T.Any,
    ) -> T.Dict[str, T.Any]:
        """One training step.

        Return a dictionary that will be passed to logging.
        """
        raise NotImplementedError

    @customer
    def visualize_train_step(