From 1f06bbf22e09842da2cbe1cd12db8ef939d614e4 Mon Sep 17 00:00:00 2001 From: Dun Liang Date: Sun, 13 Nov 2022 16:33:57 +0800 Subject: [PATCH] fix dropout2d --- python/jittor/__init__.py | 2 +- python/jittor/nn.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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).