Method test_step

climart/models/base_model.py:255–256

(self, batch: Any, batch_idx: int, dataloader_idx: int = None)

        self._start_test_epoch_time = time.time()

    def test_step(self, batch: Any, batch_idx: int, dataloader_idx: int = None):
        return self._evaluation_step(batch, batch_idx)

    def test_epoch_end(self, outputs: List[Any]) -> dict:
        test_time = time.time() - self._start_test_epoch_time