Test more intermediate batch sizes

nanoporetech · Dec 4, 2023 · 7a745e3 · 7a745e3
1 parent 03097c2
commit 7a745e3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dorado/nn/MetalCRFModel.cpp b/dorado/nn/MetalCRFModel.cpp
@@ -657,7 +657,7 @@ class MetalCaller {
             std::set<int> test_batch_sizes{min_batch_size, max_batch_size};
 
             // Add some batch sizes evenly distributed in between.
-            const int kNumIntermediateSizes = 8;
+            const int kNumIntermediateSizes = 16;
             const float test_size_increment = static_cast<float>(max_batch_size - min_batch_size) /
                                               static_cast<float>(kNumIntermediateSizes + 1);
             for (int i = 0; i < kNumIntermediateSizes; ++i) {