Skip to content
Snippets Groups Projects
Commit 84c7a7f7 authored by Daniel CARRON's avatar Daniel CARRON :b:
Browse files

Evaluate on all prediction dataloaders

parent 57b8b57d
No related branches found
No related tags found
No related merge requests found
...@@ -146,12 +146,13 @@ def evaluate( ...@@ -146,12 +146,13 @@ def evaluate(
from ..engine.evaluator import run from ..engine.evaluator import run
datamodule.prepare_data()
datamodule.setup(stage="test")
datamodule.set_chunk_size(1, 1) datamodule.set_chunk_size(1, 1)
datamodule.model_transforms = []
datamodule.prepare_data()
datamodule.setup(stage="predict")
dataloader = datamodule.test_dataloader() dataloader = datamodule.predict_dataloader()
threshold = _validate_threshold(threshold, dataloader) threshold = _validate_threshold(threshold, dataloader)
...@@ -176,7 +177,7 @@ def evaluate( ...@@ -176,7 +177,7 @@ def evaluate(
if k.startswith("_"): if k.startswith("_"):
logger.info(f"Skipping dataset '{k}' (not to be evaluated)") logger.info(f"Skipping dataset '{k}' (not to be evaluated)")
continue continue
logger.info(f"Analyzing '{threshold}' set...") logger.info(f"Analyzing '{k}' set...")
run( run(
v, v,
k, k,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment