Surrogate pinns #338

wangguan1995 · 2023-05-24T05:03:20Z

PR types

Others

PR changes

APIs

Describe

Add example of paper [Surrogate modeling for fluid flows based on physics-constrained deep learning without simulation data]

paddle-bot · 2023-05-24T05:03:24Z

Thanks for your contribution!

examples/aneurysm/aneurysm_flow.py

HydrogenSulfate · 2023-05-25T07:20:59Z

ppsci/data/dataset/array_dataset.py

+class MiniBatchDataset(io.Dataset):
+    def __init__(self, input, label, weight):
+        super().__init__()
+        self.input = input
+        self.label = label
+        self.num_samples = self.check_input(input)
+
+    def check_input(self, input):
+        len_input = set()
+        for _, value in input.items():
+            len_input.add(len(value))
+        if len(len_input) is not 1:
+            raise AttributeError("Input dimension mismatch")
+        else:
+            return list(len_input)[0]
+
+    def __getitem__(self, idx):
+        input_item = {key: value[idx] for key, value in self.input.items()}
+        label_item = {key: value[idx] for key, value in self.label.items()}
+        return (input_item, label_item)
+
+    def __len__(self):
+        return self.num_samples


weight_dict返回空字典即可，不需要写一个 MiniBatchDataset

HydrogenSulfate · 2023-05-25T07:21:06Z

ppsci/solver/printer.py

@@ -43,7 +43,7 @@ def log_train_info(trainer, batch_size, epoch_id, iter_id):

    metric_msg = ", ".join(
        [
-            f"{key}: {trainer.train_output_info[key].avg:.5f}"
+            f"{key}: {trainer.train_output_info[key].val:.9f}"


调试代码改回去

HydrogenSulfate · 2023-05-25T07:21:14Z

ppsci/solver/solver.py

-            logger.info(f"[Train][Epoch {epoch_id}/{self.epochs}][Avg] {metric_msg}")
+            # logger.info(f"[Train][Epoch {epoch_id}/{self.epochs}][Avg] {metric_msg}")


调试代码改回去

HydrogenSulfate · 2023-05-25T07:21:20Z

ppsci/solver/train.py

@@ -72,7 +72,6 @@ def train_epoch_func(solver, epoch_id: int, log_freq: int):
        if solver.update_freq > 1:
            total_loss = total_loss / solver.update_freq
        loss_dict["loss"] = float(total_loss)
-


不要删除空行

HydrogenSulfate · 2023-05-25T07:21:32Z

ppsci/solver/train.py

删除调试代码

ppsci/data/__init__.py

examples/aneurysm/aneurysm_flow.py

ppsci/arch/activation.py

examples/aneurysm/aneurysm_flow.py

HydrogenSulfate · 2023-06-03T06:16:42Z

examples/aneurysm/aneurysm_flow.py

+import matplotlib.pyplot as plt
+import numpy as np
+import paddle
+from paddle.fluid import core


HydrogenSulfate · 2023-06-03T07:11:32Z

examples/aneurysm/aneurysm_flow.py

+            bbox_inches="tight",
+        )
+        plt.close("all")
+    print(f"epochs = {EPOCHS}")


使用 ppsci.utils.logger 代替 print

from ppsci.utils import logger logger.info(...)

examples/pipe/poiseuille_flow.py

HydrogenSulfate · 2023-06-03T07:13:08Z

examples/pipe/poiseuille_flow.py

+
+    # set model
+    model_u = ppsci.arch.MLP(("sin(x)", "cos(x)", "y", "nu"), ("u"), 3, 50, "swish")
+


删除多余空行

HydrogenSulfate · 2023-06-03T07:13:43Z

ppsci/arch/activation.py

+    "swish": Swish(),
+    "silu": silu,


不要调换原来 silu 的位置

ppsci/equation/pde/navier_stokes.py

HydrogenSulfate · 2023-06-05T10:43:16Z

examples/aneurysm/aneurysm_flow.py

+    PLOT_DIR = osp.join(OUTPUT_DIR, "visu")
+    os.makedirs(PLOT_DIR, exist_ok=True)


变量定义尽量靠近使用位置

HydrogenSulfate · 2023-06-05T10:45:26Z

examples/aneurysm/aneurysm_flow.py

+    X_IN = 0
+    X_OUT = X_IN + L
+    R_INLET = 0.05
+    unique_x = np.linspace(X_IN, X_OUT, 100)


unique_x 定以后没有用到

examples/aneurysm/aneurysm_flow.py

HydrogenSulfate · 2023-06-05T10:54:57Z

examples/aneurysm/aneurysm_flow.py

+    for x0 in x_inital:
+        index = np.where(x[:, 0] == x0)[0]
+        # y is linear to scale, so we place linespace to get 1000 x, it coressponds to vessels
+        y[index] = np.linspace(-max(y_up[index]), max(y_up[index]), len(index)).reshape(


所有np.linspace显式指定dtype

examples/aneurysm/aneurysm_flow.py

examples/pipe/poiseuille_flow.py

HydrogenSulfate · 2023-06-05T11:15:32Z

examples/pipe/poiseuille_flow.py

+            lw=2.0,
+            alpha=1.0,
+        )
+        nu_current = float("{0:.5f}".format(data_1d_nu[nu_index[idxP]]))


.format全部改成f-string

HydrogenSulfate · 2023-06-05T11:17:54Z

examples/pipe/poiseuille_flow.py

+    plt.ylabel(r"PDF", fontsize=fontsize)
+    ax1.tick_params(axis="x", labelsize=fontsize)
+    ax1.tick_params(axis="y", labelsize=fontsize)
+    plt.savefig("pipe_unformUQ.png", bbox_inches="tight")


跟第一个案例一样，新建一个 PLOT_DIR，所有结果保存到这个文件里去，不要直接保存

HydrogenSulfate · 2023-06-05T11:18:06Z

ppsci/arch/activation.py

+    def __init__(self, beta: float = 1.0):
+        super().__init__()
+        self.beta = self.create_parameter(
+            shape=[1],


HydrogenSulfate · 2023-06-05T11:18:15Z

ppsci/arch/activation.py

+        super().__init__()
+        self.beta = self.create_parameter(
+            shape=[1],
+            dtype=paddle.get_default_dtype(),


删除dtype=...

ppsci/arch/activation.py

HydrogenSulfate · 2023-06-06T03:34:35Z

ppsci/arch/activation.py

+            shape=[],
+            default_initializer=paddle.nn.initializer.Constant(beta),
+        )
+        self.add_parameter("beta", self.beta)


不需要手动add_parameter，删掉

examples/pipe/poiseuille_flow.py

HydrogenSulfate · 2023-06-06T03:39:04Z

examples/aneurysm/aneurysm_flow.py

+            bbox_inches="tight",
+        )
+        plt.close("all")
+    logger.info(f"epochs = {EPOCHS}")


这句话删掉，没必要在末尾打印 EPOCHS

HydrogenSulfate · 2023-06-06T03:44:16Z

examples/aneurysm/aneurysm_flow.py

+    def single_test(x, y, scale, solver):
+        xt = paddle.to_tensor(x)
+        yt = paddle.to_tensor(y)
+        scalet = paddle.full_like(xt, scale)
+        input_dict = {"x": xt, "y": yt, "scale": scalet}
+        output_dict = solver.predict(input_dict, batch_size=100)
+        return output_dict


single_test -> model_predict

numpy的转换代码放到函数里面，return {k: v.numpy() for k, v in output_dict.items()}

给 model_predict 加上 type hint

from typing import TYPE_CHECKING if TYPE_CHECKING: from ppsci import solver ... ... def model_predict(x: np.ndarray, y: np.ndarray, scale: np.ndarray, solver: "solver.Solver"): ...

HydrogenSulfate · 2023-06-06T08:33:16Z

examples/aneurysm/aneurysm_flow.py

+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+


删除一个空行

HydrogenSulfate · 2023-06-06T08:40:59Z

examples/pipe/poiseuille_flow.py

+    for i in range(num_test):
+        u_max_a[i] = (R**2) * dP / (2 * L * data_1d_nu_distribution[i] * RHO)


改成 u_max_a = (R**2) * dP / (2 * L * data_1d_nu_distribution * RHO)

HydrogenSulfate

LGTM

* fix random and create demo * Reproduce Fig. 2.(a) successfully * reproduce fig1 (a) and (b) successfully * reproduce aneurysm case in paper 3.2 * Aneurysm converged * clean pipe flow code * add main * clean code * using xvaier instead tf net initial params * muli opt:w * need shuffle * refine by review * add wss formula * review fix * delete useless comment * reformat pipe flow * merge new NS eqn which thinks of nu as variable * reformat transform * recover mlp code * delete blank line * fix by reviews * fix epochs * delete useless lines * bug fix * forget to uncomment train * fix multi-learnable bug * fix by reviews * fix by reviews * clean activation * add reference * fix by reviews

wangguan1995 added 12 commits April 23, 2023 09:14

fix random and create demo

162a21c

Merge branch 'develop'

7b85252

Reproduce Fig. 2.(a) successfully

ca255c6

reproduce fig1 (a) and (b) successfully

cfc8407

reproduce aneurysm case in paper 3.2

774d825

Aneurysm converged

4b6c285

clean pipe flow code

e8b53e2

add main

70f77ae

clean code

200bde2

merge develop

7a0de8d

using xvaier instead tf net initial params

7d957c9

merge develop

a406cfc

HydrogenSulfate requested changes May 25, 2023

View reviewed changes

wangguan1995 added 9 commits May 31, 2023 03:02

muli opt:w

683adc7

need shuffle

9aa1279

refine by review

2cb23a3

add wss formula

5fc66e3

review fix

50d6936

delete useless comment

67f064a

merge develop

4607582

reformat pipe flow

204c5e5

merge new NS eqn which thinks of nu as variable

82e2e5d

HydrogenSulfate requested changes Jun 2, 2023

View reviewed changes

ppsci/data/__init__.py Outdated Show resolved Hide resolved

examples/aneurysm/aneurysm_flow.py Show resolved Hide resolved

wangguan1995 added 3 commits June 2, 2023 10:48

reformat transform

7bef78e

recover mlp code

334c2a9

delete blank line

b80657a

wangguan1995 requested a review from HydrogenSulfate June 2, 2023 11:02

HydrogenSulfate requested changes Jun 3, 2023

View reviewed changes

fix by reviews

ec6f9fd

wangguan1995 added 4 commits June 5, 2023 03:27

fix epochs

31aeea8

delete useless lines

669a0c5

bug fix

2af8a3a

forget to uncomment train

f821d82

wangguan1995 requested a review from HydrogenSulfate June 5, 2023 07:05

HydrogenSulfate requested changes Jun 5, 2023

View reviewed changes

fix multi-learnable bug

1f30ad4

HydrogenSulfate requested changes Jun 5, 2023

View reviewed changes

ppsci/arch/activation.py Show resolved Hide resolved

fix by reviews

169759d

wangguan1995 requested a review from HydrogenSulfate June 6, 2023 03:13

HydrogenSulfate requested changes Jun 6, 2023

View reviewed changes

wangguan1995 added 3 commits June 6, 2023 04:03

fix by reviews

620e38c

clean activation

76797fb

add reference

4d933f3

HydrogenSulfate requested changes Jun 6, 2023

View reviewed changes

wangguan1995 requested a review from HydrogenSulfate June 6, 2023 09:21

fix by reviews

e974416

HydrogenSulfate approved these changes Jun 7, 2023

View reviewed changes

HydrogenSulfate merged commit 839743b into PaddlePaddle:develop Jun 7, 2023

HydrogenSulfate deleted the surrogate_pinns branch June 7, 2023 11:59

HydrogenSulfate restored the surrogate_pinns branch September 20, 2023 03:36

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Surrogate pinns #338

Surrogate pinns #338

wangguan1995 commented May 24, 2023

paddle-bot bot commented May 24, 2023

HydrogenSulfate May 25, 2023

HydrogenSulfate May 25, 2023

HydrogenSulfate May 25, 2023

HydrogenSulfate May 25, 2023

HydrogenSulfate May 25, 2023

HydrogenSulfate Jun 3, 2023

HydrogenSulfate Jun 3, 2023

HydrogenSulfate Jun 3, 2023

HydrogenSulfate Jun 3, 2023

HydrogenSulfate Jun 5, 2023

HydrogenSulfate Jun 5, 2023

HydrogenSulfate Jun 5, 2023

HydrogenSulfate Jun 5, 2023

HydrogenSulfate Jun 5, 2023

HydrogenSulfate Jun 5, 2023

HydrogenSulfate Jun 5, 2023

HydrogenSulfate Jun 6, 2023

HydrogenSulfate Jun 6, 2023

HydrogenSulfate Jun 6, 2023

HydrogenSulfate Jun 6, 2023

HydrogenSulfate Jun 6, 2023

HydrogenSulfate left a comment

		logger.info(f"[Train][Epoch {epoch_id}/{self.epochs}][Avg] {metric_msg}")
		# logger.info(f"[Train][Epoch {epoch_id}/{self.epochs}][Avg] {metric_msg}")


		# set model
		model_u = ppsci.arch.MLP(("sin(x)", "cos(x)", "y", "nu"), ("u"), 3, 50, "swish")

		PLOT_DIR = osp.join(OUTPUT_DIR, "visu")
		os.makedirs(PLOT_DIR, exist_ok=True)

		# See the License for the specific language governing permissions and
		# limitations under the License.

		for i in range(num_test):
		u_max_a[i] = (R*2) dP / (2 * L * data_1d_nu_distribution[i] * RHO)

Surrogate pinns #338

Surrogate pinns #338

Conversation

wangguan1995 commented May 24, 2023

PR types

PR changes

Describe

paddle-bot bot commented May 24, 2023

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

HydrogenSulfate left a comment

Choose a reason for hiding this comment