diff --git a/python/jittor/__init__.py b/python/jittor/__init__.py index f96fab76..6ed246b2 100644 --- a/python/jittor/__init__.py +++ b/python/jittor/__init__.py @@ -9,7 +9,7 @@ # file 'LICENSE.txt', which is part of this source code package. # *************************************************************** -__version__ = '1.3.5.30' +__version__ = '1.3.5.32' from jittor_utils import lock with lock.lock_scope(): ori_int = int diff --git a/python/jittor/nn.py b/python/jittor/nn.py index 2112c1e6..c21251d2 100644 --- a/python/jittor/nn.py +++ b/python/jittor/nn.py @@ -572,7 +572,7 @@ class Dropout2d(Module): return output def dropout2d(x,p=0.5,is_train=False): - return Dropout(p=p,is_train=is_train)(x) + return Dropout2d(p=p,is_train=is_train)(x) class DropPath(Module): '''Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks).