Skip to content

Commit 6df895b

Browse files
authored
fix: use register_buffer for kernel and kernel_vol in LocalNormalizedCrossCorrelationLoss
Refactor kernel initialization to use register_buffer for better state management. Signed-off-by: Zeeshan Modi <92383127+Zeesejo@users.noreply.github.com>
1 parent cc92126 commit 6df895b

1 file changed

Lines changed: 2 additions & 3 deletions

File tree

monai/losses/image_dissimilarity.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -111,9 +111,8 @@ def __init__(
111111
raise ValueError(f"kernel_size must be odd, got {self.kernel_size}")
112112

113113
_kernel = look_up_option(kernel_type, kernel_dict)
114-
self.kernel = _kernel(self.kernel_size)
115-
self.kernel.require_grads = False
116-
self.kernel_vol = self.get_kernel_vol()
114+
self.register_buffer("kernel", _kernel(self.kernel_size))
115+
self.register_buffer("kernel_vol", self.get_kernel_vol())
117116

118117
self.smooth_nr = float(smooth_nr)
119118
self.smooth_dr = float(smooth_dr)

0 commit comments

Comments
 (0)