Skip to content

Commit 54d3224

Browse files
authored
Update monai/losses/image_dissimilarity.py
Signed-off-by: Eric Kerfoot <17726042+ericspod@users.noreply.github.com>
1 parent 87aa8a2 commit 54d3224

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

monai/losses/image_dissimilarity.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -111,6 +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: torch.Tensor
115+
self.kernel_vol: torch.Tensor
114116
self.register_buffer("kernel", _kernel(self.kernel_size), persistent=False)
115117
self.register_buffer("kernel_vol", self.get_kernel_vol(), persistent=False)
116118

0 commit comments

Comments
 (0)