From e8a20a0ba4fa62d48e732194485cecd525b94ac8 Mon Sep 17 00:00:00 2001
From: NicolasSR <54904691+NicolasSR@users.noreply.github.com>
Date: Tue, 30 Jan 2024 11:05:43 +0100
Subject: [PATCH] Update demo_rom_manager_nn.py

---
 rom_application/RomManager_cantilever_NN/demo_rom_manager_nn.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/rom_application/RomManager_cantilever_NN/demo_rom_manager_nn.py b/rom_application/RomManager_cantilever_NN/demo_rom_manager_nn.py
index d6cfbb30..2082767f 100644
--- a/rom_application/RomManager_cantilever_NN/demo_rom_manager_nn.py
+++ b/rom_application/RomManager_cantilever_NN/demo_rom_manager_nn.py
@@ -85,7 +85,7 @@ def GetRomManagerParameters():
     general_rom_manager_parameters = KratosMultiphysics.Parameters("""{
             "rom_stages_to_train" : ["ROM"],             // ["ROM","HROM"]
             "rom_stages_to_test" : [],              // ["ROM","HROM"]
-            "rom_stages_nn_fit"  : ["GenValData","GenTrainData","TrainNN"],      // ["GenValData","GenTrainData","TrainNN"]
+	    "type_of_decoder" : "ann_enhanced",           // "linear" "ann_enhanced",  TODO: add "quadratic"
             "paralellism" : null,                        // null, TODO: add "compss"
             "projection_strategy": "galerkin",            // "lspg", "galerkin", "petrov_galerkin"
             "assembling_strategy": "global",            // "global", "elemental"