Skip to content

Commit 3d599f9

Browse files
Adjust auto scaling of FC0 to not go under 6 as otherwise clamping would sometimes lead to kernel sizes that were not divisible by 3 (e.g. 4).
1 parent 37b098a commit 3d599f9

1 file changed

Lines changed: 1 addition & 3 deletions

File tree

src/eir/models/input/array/models_locally_connected.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -838,9 +838,7 @@ def _clamp_kernel_for_min_chunks(
838838

839839

840840
def _get_auto_scaled_fc0_kernel(n_snps: int) -> int:
841-
if n_snps < 1_000:
842-
return 3
843-
elif n_snps < 10_000:
841+
if n_snps < 1_000 or n_snps < 10_000:
844842
return 6
845843
elif n_snps < 100_000:
846844
return 12

0 commit comments

Comments
 (0)