From 51861034425e27780030109cc4649382a2d04637 Mon Sep 17 00:00:00 2001
From: co63oc <co63@163.com>
Date: Thu, 14 Sep 2023 07:28:20 +0800
Subject: [PATCH] Fix

---
 jointContribution/PIRBN/pirbn.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/jointContribution/PIRBN/pirbn.py b/jointContribution/PIRBN/pirbn.py
index 8acacd442a..02f5c86ee3 100644
--- 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)