Fixed a bug for DAReg. (#649)

friedenhe · web-flow · commit 3492f59c45d6 · 2024-06-18T13:02:57.000-05:00
diff --git a/src/adjoint/DARegression/DARegression.C b/src/adjoint/DARegression/DARegression.C
@@ -592,7 +592,7 @@ label DARegression::compute()
                 // loop over all cells
                 forAll(features_[modelName][idxI], cellI)
                 {
-                    counterI = cellI * nCells + idxI;
+                    counterI = cellI * nInputs + idxI;
                     featuresFlattenArray_[counterI] = features_[modelName][idxI][cellI];
                 }
             }
@@ -672,10 +672,12 @@ label DARegression::nParameters(word modelName)
     else if (modelType_[modelName] == "externalTensorFlow")
     {
         // do nothing
+        return 0;
     }
     else
     {
         FatalErrorIn("") << "modelType_: " << modelType_[modelName] << " not supported. Options are: neuralNetwork, radialBasisFunction, and externalTensorFlow" << abort(FatalError);
+        return 0;
     }
 }
 

Original file line number	Diff line number	Diff line change
`@@ -592,7 +592,7 @@ label DARegression::compute()`
`592`	`592`	`// loop over all cells`
`593`	`593`	`forAll(features_[modelName][idxI], cellI)`
`594`	`594`	`{`
`595`		`- counterI = cellI * nCells + idxI;`
	`595`	`+ counterI = cellI * nInputs + idxI;`
`596`	`596`	`featuresFlattenArray_[counterI] = features_[modelName][idxI][cellI];`
`597`	`597`	`}`
`598`	`598`	`}`
`@@ -672,10 +672,12 @@ label DARegression::nParameters(word modelName)`
`672`	`672`	`else if (modelType_[modelName] == "externalTensorFlow")`
`673`	`673`	`{`
`674`	`674`	`// do nothing`
	`675`	`+ return 0;`
`675`	`676`	`}`
`676`	`677`	`else`
`677`	`678`	`{`
`678`	`679`	`FatalErrorIn("") << "modelType_: " << modelType_[modelName] << " not supported. Options are: neuralNetwork, radialBasisFunction, and externalTensorFlow" << abort(FatalError);`
	`680`	`+ return 0;`
`679`	`681`	`}`
`680`	`682`	`}`
`681`	`683`