diff --git a/src/descriptor_identifier/solver/SISSOSolver.cpp b/src/descriptor_identifier/solver/SISSOSolver.cpp
index 46acf4462d05276be44631434ead5fb5b1edae95..6c3da19bcf65cf2dc845ff224ea94f1619fb05df 100644
--- a/src/descriptor_identifier/solver/SISSOSolver.cpp
+++ b/src/descriptor_identifier/solver/SISSOSolver.cpp
@@ -122,7 +122,7 @@ void SISSOSolver::l0_regularization(const int n_dim)
 
     while (!feature_combinations.is_finished())
     {
-        const size_t MAX_BATCH_SIZE = 128;
+        const size_t MAX_BATCH_SIZE = 4096;
         std::vector<std::vector<int>> feature_indices;
         for (auto counter = 0; counter < MAX_BATCH_SIZE; ++counter)
         {
diff --git a/src/loss_function/LossFunctionPearsonRMSE.hpp b/src/loss_function/LossFunctionPearsonRMSE.hpp
index 0b7757aab6928f2a645e7b5488a0913c362c4b6e..48dc92fe84f99207921e2aaf1fd6e757955c65ab 100644
--- a/src/loss_function/LossFunctionPearsonRMSE.hpp
+++ b/src/loss_function/LossFunctionPearsonRMSE.hpp
@@ -42,7 +42,7 @@ public:
     using PropertyView = Kokkos::View<double**, Kokkos::LayoutLeft>;
 
 protected:
-    static constexpr int MAX_BATCHES = 128;
+    static constexpr int MAX_BATCHES = 4096;
 
     /// dim 0: material samples
     /// dim 1: features