diff --git a/src/dfChemistryModel/dfChemistryModel.C b/src/dfChemistryModel/dfChemistryModel.C
index 334be4d0..8ac219dc 100644
--- a/src/dfChemistryModel/dfChemistryModel.C
+++ b/src/dfChemistryModel/dfChemistryModel.C
@@ -83,6 +83,19 @@ Foam::dfChemistryModel<ThermoType>::dfChemistryModel
         mesh_,
         dimensionedScalar(dimEnergy/dimVolume/dimTime, 0)
     ),
+    selectDNN_
+    (
+        IOobject
+        (
+            "selectDNN",
+            mesh_.time().timeName(),
+            mesh_,
+            IOobject::NO_READ,
+            IOobject::AUTO_WRITE
+        ),
+        mesh_,
+        dimensionedScalar(dimless, -1)
+    ),
     balancer_(createBalancer()),
     cpuTimes_
     (
@@ -522,18 +535,21 @@ Foam::dfChemistryModel<ThermoType>::getGPUProblems
         {
             problem.DNNid = 0;
             problemList.append(problem);
+            selectDNN_[cellI]=0;
             continue;
         }
         if(((Qdot_[cellI] >= Qdotact2_) && (T_[cellI] < Tact2_)&&(T_[cellI] >= Tact1_))||((Qdot_[cellI] > Qdotact3_) && T_[cellI] > Tact2_))  //choose2
         {
             problem.DNNid = 1;
             problemList.append(problem);
+            selectDNN_[cellI]=1;
             continue;
         }
         if  ((Qdot_[cellI] < Qdotact3_) && (T_[cellI] >= Tact2_) && (Qdot_[cellI]!=0)) //choose3
         {
             problem.DNNid = 2;
             problemList.append(problem);
+            selectDNN_[cellI]=2;
             continue;
         }
 
diff --git a/src/dfChemistryModel/dfChemistryModel.H b/src/dfChemistryModel/dfChemistryModel.H
index 369f107e..75fab9a7 100644
--- a/src/dfChemistryModel/dfChemistryModel.H
+++ b/src/dfChemistryModel/dfChemistryModel.H
@@ -125,7 +125,8 @@ public IOdictionary
     volScalarField& psi_;
     // heat release rate, [J/m^3/s]
     volScalarField Qdot_;
-
+    // DNN selection fields 
+    volScalarField selectDNN_;
 
 #ifdef USE_LIBTORCH
     Switch torchSwitch_;