fix bugs on NMDA synapses

brainpy · Apr 22, 2022 · fe95783 · fe95783
1 parent 1c2df42
commit fe95783
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/brainpy/dyn/synapses/abstract_models.py b/brainpy/dyn/synapses/abstract_models.py
@@ -1227,8 +1227,9 @@ def update(self, _t, _dt):
         post_g = bm.sum(self.g)
         if not self.conn.include_self:
           post_g = post_g - self.g
+        post_g = post_g * self.g_max
       else:
-        post_g = self.g * self.g_max
+        post_g = self.g @ self.g_max
     elif isinstance(self.conn, One2One):
       post_g = self.g_max * self.g
     else:
@@ -1242,4 +1243,4 @@ def update(self, _t, _dt):
 
     # output
     g_inf = 1 + self.cc_Mg / self.beta * bm.exp(-self.alpha * self.post.V)
-    self.post.input -= post_g * (self.post.V - self.E) / g_inf
+    self.post.input += post_g * (self.E - self.post.V) / g_inf