Skip to content

Commit 24acce2

Browse files
committed
change fine_spv_max to 500 to avoid bottleneck
1 parent 173d448 commit 24acce2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/jamma/matching_module.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -285,7 +285,7 @@ def __init__(self, config, profiler):
285285
self.subpixel_mlp = nn.Sequential(nn.Linear(2 * dim_f, 2 * dim_f, bias=False),
286286
nn.ReLU(),
287287
nn.Linear(2 * dim_f, 4, bias=False))
288-
self.fine_spv_max = None # saving memory
288+
self.fine_spv_max = 500 # saving memory
289289
self.profiler = profiler
290290

291291
def forward(self, feat_f0_unfold, feat_f1_unfold, data):

0 commit comments

Comments
 (0)