Fix

PaddlePaddle · Sep 13, 2023 · 5186103 · 5186103
1 parent 265376e
commit 5186103
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/jointContribution/PIRBN/pirbn.py b/jointContribution/PIRBN/pirbn.py
@@ -48,7 +48,7 @@ def forward(self, input_data):
         return [u_xx, u_b]
 
     def cal_ntk(self, x):
-        # Page5, \gamma variable
+        # Formula (4), Page5, \gamma variable
         gamma_g = 0.0
         gamma_b = 0.0
         n_neu = self.rbn.n_neu
@@ -63,7 +63,7 @@ def cal_ntk(self, x):
                 gamma_g = gamma_g + paddle.sum(j**2) / n1
             temp = paddle.concat((l1t[0], l1t[1].reshape((1, n_neu))), axis=1)
             if i == 0:
-                # Page8, Kg variable
+                # Fig.1, Page8, Kg variable
                 Kg = temp
             else:
                 Kg = paddle.concat((Kg, temp), axis=0)