Skip to content

Commit

Permalink
remove model parameters in predict
Browse files Browse the repository at this point in the history
  • Loading branch information
dyollb committed Nov 7, 2024
1 parent 288663f commit 3645cab
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions src/segmantic/seg/monai_unet.py
Original file line number Diff line number Diff line change
Expand Up @@ -554,9 +554,6 @@ def predict(
test_labels: Optional[list[Path]] = None,
output_dir: Path = None,
tissue_dict: dict[str, int] = None,
channels: tuple[int, ...] = (16, 32, 64, 128, 256),
strides: tuple[int, ...] = (2, 2, 2, 2),
dropout: float = 0.0,
spacing: Sequence[float] = [],
gpu_ids: list[int] = [],
) -> None:
Expand All @@ -569,7 +566,7 @@ def predict(
net: Net = Net.load_from_checkpoint(f"{model_file}", **settings)
else:
net = Net.load_from_checkpoint(
f"{model_file}", channels=channels, strides=strides, dropout=dropout
f"{model_file}"
)
num_classes = net.num_classes

Expand Down

0 comments on commit 3645cab

Please sign in to comment.