Tutorials/compressible_flow/NICFD_nozzle/PhysicsInformed/2_train_PINN.py at 8205ce14e570d18e965f5174e57716c5cbc34789 · su2code/Tutorials · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
###############################################################################################
#       #      _____ __  _____      ____        __        __  ____                   #        #
#       #     / ___// / / /__ \    / __ \____ _/ /_____ _/  |/  (_)___  ___  _____   #        #
#       #     \__ \/ / / /__/ /   / / / / __ `/ __/ __ `/ /|_/ / / __ \/ _ \/ ___/   #        #
#       #    ___/ / /_/ // __/   / /_/ / /_/ / /_/ /_/ / /  / / / / / /  __/ /       #        #
#       #   /____/\____//____/  /_____/\__,_/\__/\__,_/_/  /_/_/_/ /_/\___/_/        #        #
#       #                                                                            #        #
###############################################################################################

################################ FILE NAME: 2_train_PINN.py ###################################
#=============================================================================================#
# author: Evert Bunschoten                                                                    |
#    :PhD Candidate ,                                                                         |
#    :Flight Power and Propulsion                                                             |
#    :TU Delft,                                                                               |
#    :The Netherlands                                                                         |
#                                                                                             |
#                                                                                             |
# Description:                                                                                |
#  Initate physics-informed machine learning process for training the neural network used to  |
#  model the fluid properties of siloxane MM in NICFD with the SU2 data-driven fluid model.   |
#                                                                                             |
# Version: 2.0.0                                                                              |
#                                                                                             |
#=============================================================================================#

from su2dataminer.config import Config_NICFD
from su2dataminer.manifold import TrainMLP_NICFD

# Load SU2 DataMiner configuration.
Config = Config_NICFD("SU2DataMiner_MM.cfg")

# Set learning rate parameters and define network architecture.
Eval = TrainMLP_NICFD(Config)

# Initial learning rate: 10^-3, learning rate decay parameter: 9.8787, mini-batch size: 2^6.
Eval.SetAlphaExpo(-3.0)
Eval.SetLRDecay(+9.8787e-01)
Eval.SetBatchExpo(6)

# Network architecture: two hidden layers with 12 nodes.
Eval.SetHiddenLayers([12,12])
# Hidden layer activation function: exp(x)
Eval.SetActivationFunction("exponential")
# Display training progress in the terminal.
Eval.SetVerbose(1)

# Initiate training process.
Eval.CommenceTraining()
Eval.TrainPostprocessing()

Config.UpdateMLPHyperParams(Eval)
Config.SaveConfig()