PINN代码实例-顶盖驱动流问题

WangQi 0

http://dyfluid.com/pinn3.html 李老师您好，我想请问两个问题：
对于这段代码：

auto cont1 = dudx + dvdy;
        auto mom1 = 2.0*u_pred*dudx + v_pred*dudy + u_pred*dvdy 
          - nu*(dudxx + dudyy) - nu*(dudxx + dvdxy) + dpdx;
        auto mom2 = u_pred*dvdx + v_pred*dudx + 2.0*v_pred*dvdy 
          - nu*(dvdxx + dvdyy) - nu*(dudxy + dvdyy) + dpdy;
        auto loss_cont1 = crit(cont1, 0*cont1);
        auto loss_mom1 = crit(mom1, 0*mom1);
        auto loss_mom2 = crit(mom2, 0*mom2);

mom1中u_pred*dudx为何要*2
粘度项为何写成 - nu*(dudxx + dudyy) - nu*(dudxx + dvdxy)，这好像并不太符合NS方程中粘度项沿X方向的分解？

请李老师指点一下

李东岳

动量方程有不同的形式，我在植入的时候，发现不同的形式对训练的快慢有影响，蜗植入了某种形式，具体公式我现在没有了，我一会更新到笔记里面。

当然也不排除是bug的情况。

李东岳

图1.JPG

公式就是上面这个。感兴趣你可以测试一下不同的方程形式对训练的影响。

小菜鸟

http://dyfluid.com/pinn3.html
李老师您好，借楼请教个问题：
对于这段代码：

    auto p_pred = UP.index({torch::indexing::Slice(), 2});

    auto loss_dataP = crit(p_pred[0], 0*p_pred[0]);
    auto loss_data = 100*(loss_dataP);

1、p_pred是网络输出结果中的slice切片后的一维张量，为什么只对p_pred取第一个数据计算损失mse_loss？

请李老师指点一下

小菜鸟

@李东岳李老师，这个（3.95）和（3.96）两个方程左边的第一项有错误，第一项应该还需要乘一个u1, u2。

李东岳

是个bug，你直接把loss_dataP删了就行，这个应该是我当时要增加数据点用的。目前没有作用了。loss_data可以删了，目前跟loss_pde重复

怎么样，这部分玩的怎么样，有什么感受。PINN跟FVM是不是大不一样

CFD中文网

PINN代码实例-顶盖驱动流问题