levimcclenny / SA-PINNs

Implementation of the paper "Self-Adaptive Physics-Informed Neural Networks using a Soft Attention Mechanism" [AAAI-MLPS 2021]
https://arxiv.org/abs/2009.04544
163 stars 36 forks source link

Is there a wrong? #8

Open usermtl opened 1 year ago

usermtl commented 1 year ago

mse_0_u = tf.reduce_mean(tf.square(u_weights(u0 - u0_pred))) mse_f_u = tf.reduce_mean(tf.square(col_weightsf_u_pred))

order your paper, image

mse_0_u and mse_f_u should be:

mse_0_u = tf.reduce_mean(u_weightstf.square((u0 - u0_pred))) mse_f_u = tf.reduce_mean(col_weightstf.square(f_u_pred))

I am not sure for this, is it right?