From 692a1be676f010bda63f227dd1cbb5632c18f2ef Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Wed, 26 Jun 2024 19:51:03 +0200
Subject: [PATCH] [libs.segmentation.engine.predictor] Fix documentation
---
src/mednet/libs/segmentation/engine/predictor.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/src/mednet/libs/segmentation/engine/predictor.py b/src/mednet/libs/segmentation/engine/predictor.py
index ecac9359..a33a3b80 100644
--- a/src/mednet/libs/segmentation/engine/predictor.py
+++ b/src/mednet/libs/segmentation/engine/predictor.py
@@ -134,11 +134,11 @@ def run(
model
Neural network model (e.g. pasa).
datamodule
- The lightning DataModule to use for training **and** validation.
+ The lightning DataModule to run predictions on.
device_manager
- An internal device representation, to be used for training and
- validation. This representation can be converted into a pytorch device
- or a lightning accelerator setup.
+ An internal device representation, to be used for prediction. This
+ representation can be converted into a pytorch device or a lightning
+ accelerator setup.
output_folder
Folder where to store HDF5 representations of probability maps.
--
GitLab