Added name of model in the training config parameters

KratosMultiphysics · Jan 29, 2024 · f219f68 · f219f68
1 parent ec0629c
commit f219f68
Showing 1 changed file with 3 additions and 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
@@ -130,7 +130,9 @@ def GetRomManagerParameters():
 		        "validation_set": "rom_data/SnapshotsMatrices/fom_snapshots_val.npy",
 		        "phi_matrix": "rom_data/RightBasisMatrix.npy",
 		        "sigma_vector": "rom_data/SingularValuesVector.npy"
-		    }
+		    },
+		    "use_automatic_name": true,
+		    "custom_name": "test_neural_network"
                 },
                 "online":{
                     "model_name": "NN_model_3.10_[2](200,200)_lrsgdr.0.001_batchsize4"