src/poet.cpp

@@ -158,7 +158,7 @@ int parseInitValues(int argc, char **argv, RuntimeParameters &params) {
                             "Enable AI surrogate for chemistry module");
   ai_option_group
       ->add_option("--ai-backend", params.ai_backend,
-                   "Desired ai backend (1: python (keras), 2: naa)")
+                   "Desired ai backend (0: python (keras), 1: naa, 2: cuda)")
       ->check(CLI::PositiveNumber)
       ->default_val(RuntimeParameters::AI_BACKEND_DEFAULT);
 
@@ -386,10 +386,10 @@ static Rcpp::List RunMasterLoop(RInsidePOET &R, const RuntimeParameters &params,
     //    data frame with elements that are used as ai outputs for the
     //    retraining step.
 
-    chem.getField().update(diffusion.getField());
-
     if (params.ai || params.copy_non_reactive_regions) {
 
+      chem.getField().update(diffusion.getField());
+
       R["TMP"] = Rcpp::wrap(chem.getField().AsVector());
       R.parseEval(
           std::string("field <- setNames(data.frame(matrix(TMP, nrow=" +