lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix dropout2d

This commit is contained in:
Dun Liang 2022-11-13 16:33:57 +08:00
parent 74932f3c32
commit 1f06bbf22e
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/jittor/__init__.py
View File

@ -9,7 +9,7 @@
# file 'LICENSE.txt', which is part of this source code package. # 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 from jittor_utils import lock
with lock.lock_scope(): with lock.lock_scope():
ori_int = int ori_int = int

2
python/jittor/nn.py
View File

@ -572,7 +572,7 @@ class Dropout2d(Module):
return output return output
def dropout2d(x,p=0.5,is_train=False): 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): class DropPath(Module):
'''Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks). '''Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks).