Merge pull request #384 from minzhang0929/master

update pytorchFunctions
deepmodeling · Dec 21, 2023 · 15ae774 · 15ae774
2 parents 70441ad + 4188c62
commit 15ae774
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/dfChemistryModel/pytorchFunctions.H b/src/dfChemistryModel/pytorchFunctions.H
@@ -145,9 +145,9 @@ Foam::scalar Foam::dfChemistryModel<ThermoType>::solve_DNN(const DeltaTType &del
 
             /*=============================construct solutions=============================*/
             std::chrono::steady_clock::time_point start6 = std::chrono::steady_clock::now();
-            std::vector<double> outputsVec0(star, star+outputLength[0] * 7); //the float number is sample_length*sample_number
-            std::vector<double> outputsVec1(star+outputLength[0] * 7, star+outputLength[1] * 7);
-            std::vector<double> outputsVec2(star+outputLength[1] * 7, star+outputLength[2] * 7);
+            std::vector<double> outputsVec0(star, star+outputLength[0] * mixture_.nSpecies()); //the float number is sample_length*sample_number
+            std::vector<double> outputsVec1(star+outputLength[0] * mixture_.nSpecies(), star+outputLength[1] * mixture_.nSpecies());
+            std::vector<double> outputsVec2(star+outputLength[1] * mixture_.nSpecies(), star+outputLength[2] * mixture_.nSpecies());
             std::vector<std::vector<double>> results = {outputsVec0, outputsVec1, outputsVec2};
             updateSolutionBuffer(solutionBuffer, results, cellIDBuffer, problemCounter);
             std::chrono::steady_clock::time_point stop6 = std::chrono::steady_clock::now();