Method test_step

pointersect/script/train_v2.py:1322–1338

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

        return all_loss_dict

    def test_step(
            self,
            epoch: int,
            bidx: int,
            batch: T.Any,
    ):
        self.reset_losses()
        self.reset_outputs()

        with torch.no_grad():
            all_loss_dict = self._step(
                epoch=epoch,
                bidx=bidx,
                batch=batch,
                update=False,
            )
        return all_loss_dict

    def visualize_train_step(
            self,