Method visualize_test_step

cdslib/core/script/base_train.py:1182–1198

(
            self,
            epoch: int,
            total_batch_count: int,
            bidx: int,
            batch: T.Any,
            out_dict: T.Dict[str, T.Any],
            logger: Logger,
            batch_time: float,
            step_time: float,
            epoch_time: float,
    )

    @customer
    def visualize_test_step(
            self,
            epoch: int,
            total_batch_count: int,
            bidx: int,
            batch: T.Any,
            out_dict: T.Dict[str, T.Any],
            logger: Logger,
            batch_time: float,
            step_time: float,
            epoch_time: float,
    ):
        self.logger.info(
            f"batch uses: {batch_time:.4f} secs, "
            f"(step uses: {step_time:.4f} secs, "
            f"{step_time / batch_time * 100.:.2f}%)"
        )

    @customer
    def finish_procedure(self):