Skip to content

Commit

Permalink
made some code changes in dynunet_wrapper
Browse files Browse the repository at this point in the history
- change the spacing -> spacing_for_internal_computations
- change the code to ensure that it only runs through compute when needed
  • Loading branch information
benmalef committed Oct 4, 2024
1 parent d6f1d64 commit 6468f12
Showing 1 changed file with 10 additions and 4 deletions.
14 changes: 10 additions & 4 deletions GANDLF/models/dynunet_wrapper.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,13 +64,19 @@ def __init__(self, parameters: dict):

patch_size = parameters.get("patch_size", None)
spacing = parameters.get(
"spacing", [1.0 for i in range(parameters["model"]["dimension"])]
"spacing_for_internal_computations", [1.0 for i in range(parameters["model"]["dimension"])]
)
kernel_size, strides = get_kernels_strides(patch_size, spacing)
parameters["model"]["kernel_size"] = parameters["model"].get(
"kernel_size", kernel_size
"kernel_size", None
)
parameters["model"]["strides"] = parameters["model"].get("strides", strides)
parameters["model"]["strides"] = parameters["model"].get("strides", None)
if (parameters["model"]["kernel_size"] is None) or (
parameters["model"]["strides"] is None
):
kernel_size, strides = get_kernels_strides(patch_size, spacing)
parameters["model"]["kernel_size"] = kernel_size
parameters["model"]["strides"] = strides

parameters["model"]["filters"] = parameters["model"].get("filters", None)
parameters["model"]["act_name"] = parameters["model"].get(
"act_name", ("leakyrelu", {"inplace": True, "negative_slope": 0.01})
Expand Down

0 comments on commit 6468f12

Please sign in to comment.