增加基类的描述字符

fangwei123456 · Sep 14, 2020 · e748c64 · e748c64
1 parent 332ccf0
commit e748c64
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/spikingjelly/clock_driven/surrogate.py b/spikingjelly/clock_driven/surrogate.py
@@ -59,6 +59,9 @@ def __init__(self, alpha, spiking=True, learnable=False):
         else:
             self.f = self.primitive_function
 
+    def extra_repr(self):
+        return f'alpha={self.alpha}, spiking={self.spiking}, learnable={self.alpha}'
+
     @staticmethod
     def spiking_function(x, alpha):
         raise NotImplementedError