增加ATAN的默认alpha

fangwei123456 · Sep 14, 2020 · 9ce6303 · 9ce6303
1 parent afe1f7f
commit 9ce6303
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/spikingjelly/clock_driven/surrogate.py b/spikingjelly/clock_driven/surrogate.py
@@ -684,6 +684,8 @@ def backward(ctx, grad_output):
 
 
 class ATan(SurrogateFunctionBase):
+    def __init__(self, alpha=2.0, spiking=True, learnable=False):
+        super().__init__(alpha, spiking, learnable)
     '''
     * :ref:`API in English <ATan.__init__-en>`
     .. _ATan.__init__-cn:

diff --git a/spikingjelly/datasets/__init__.py b/spikingjelly/datasets/__init__.py
@@ -1,11 +1,11 @@
 from torch.utils.data import Dataset
-import tqdm
 import os
 import numpy as np
 import torch
 import threading
 import zipfile
 from torchvision.datasets import utils
+
 class FunctionThread(threading.Thread):
     def __init__(self, f, *args, **kwargs):
         super().__init__()