Skip to content

Commit e111ef9

Browse files
authored
Update demo_rom_manager_nn.py
Added comments on ROM NN parameters
1 parent 39f0b4a commit e111ef9

1 file changed

Lines changed: 9 additions & 6 deletions

File tree

rom_application/RomManager_cantilever_NN/demo_rom_manager_nn.py

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -116,14 +116,17 @@ def GetRomManagerParameters():
116116
"ann_enhanced_settings":{
117117
"saved_models_root_path": "rom_data/saved_nn_models/",
118118
"training":{
119-
"modes":[3,10],
120-
"layers_size":[200,200],
119+
"modes":[
120+
3, // Highest mode to be used as inferior modes (size of NN input)
121+
10 // Highest mode to be used as superior modes (size of NN output + size of NN input)
122+
],
123+
"layers_size":[200,200], // Size of each hidden layer in the Neural Network (for more layers, append more values)
121124
"batch_size":4,
122125
"epochs":10,
123-
"lr_strategy": {
124-
"scheduler": "sgdr", // "const", "steps", "sgdr"
125-
"base_lr": 0.001,
126-
"additional_params": [1e-4, 10, 400] // const:[], steps/sgdr:["min_lr", "reduction_factor","update_period"]
126+
"lr_strategy": { // Learning Rate update strategy
127+
"scheduler": "sgdr", // "const", "steps", "sgdr"
128+
"base_lr": 0.001, // Initial LR
129+
"additional_params": [1e-4, 10, 400] // 'const' -> []; 'steps' or 'sgdr' -> [minimum_LR, reduction_factor, period_length]
127130
},
128131
"database":{
129132
"training_set": "rom_data/SnapshotsMatrices/fom_snapshots.npy",

0 commit comments

Comments
 (0)