wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-12 12:40:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

make evaluate private (#1260)

Browse Source 
* make evaluate private

* changelog
This commit is contained in:
Jirka Borovec
2020-03-30 12:14:27 -04:00
committed by GitHub
parent 6dfe9951e1
commit c869dd8b8f
4 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/trainer/evaluation_loop.py
+2 -2
View File
@@ -217,7 +217,7 @@ class TrainerEvaluationLoopMixin(ABC):
def reset_val_dataloader(self, *args):
"""Warning: this is just empty shell for code implemented in other class."""
def evaluate(self, model: LightningModule, dataloaders, max_batches: int, test_mode: bool = False):
def _evaluate(self, model: LightningModule, dataloaders, max_batches: int, test_mode: bool = False):
"""Run evaluation code.
Args:
@@ -365,7 +365,7 @@ class TrainerEvaluationLoopMixin(ABC):
setattr(self, f'{"test" if test_mode else "val"}_progress_bar', pbar)
# run evaluation
eval_results = self.evaluate(self.model, dataloaders, max_batches, test_mode)
eval_results = self._evaluate(self.model, dataloaders, max_batches, test_mode)
_, prog_bar_metrics, log_metrics, callback_metrics, _ = self.process_output(
eval_results)