Commit 6df895b
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
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
111 | 111 | | |
112 | 112 | | |
113 | 113 | | |
114 | | - | |
115 | | - | |
116 | | - | |
| 114 | + | |
| 115 | + | |
117 | 116 | | |
118 | 117 | | |
119 | 118 | | |
| |||
0 commit comments