Skip to content
Snippets Groups Projects
Commit 692a1be6 authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

[libs.segmentation.engine.predictor] Fix documentation

parent e5f0b7ac
No related branches found
No related tags found
1 merge request!46Create common library
Hide whitespace changes
Inline Side-by-side
src/mednet/libs/segmentation/engine/predictor.py
+ 4
− 4
View file @ 692a1be6
...@@ -134,11 +134,11 @@ def run( ...@@ -134,11 +134,11 @@ def run(
model model
Neural network model (e.g. pasa). Neural network model (e.g. pasa).
datamodule datamodule
The lightning DataModule to use for training **and** validation. The lightning DataModule to run predictions on.
device_manager device_manager
An internal device representation, to be used for training and An internal device representation, to be used for prediction. This
validation. This representation can be converted into a pytorch device representation can be converted into a pytorch device or a lightning
or a lightning accelerator setup. accelerator setup.
output_folder output_folder
Folder where to store HDF5 representations of probability maps. Folder where to store HDF5 representations of probability maps.
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment