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

polish error msg

This commit is contained in:
Dun Liang 2021-02-03 11:55:36 +08:00
parent 5b8d2fe13c
commit ed1738248e
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/jittor/__init__.py
View File

@ -8,7 +8,7 @@
# This file is subject to the terms and conditions defined in # This file is subject to the terms and conditions defined in
# file 'LICENSE.txt', which is part of this source code package. # file 'LICENSE.txt', which is part of this source code package.
# *************************************************************** # ***************************************************************
__version__ = '1.2.2.27' __version__ = '1.2.2.28'
from . import lock from . import lock
with lock.lock_scope(): with lock.lock_scope():
ori_int = int ori_int = int

2
python/jittor/dataset/dataset.py
View File

@ -392,7 +392,7 @@ Example::
self.real_len = len(index_list) self.real_len = len(index_list)
self.real_batch_size = real_batch_size self.real_batch_size = real_batch_size
assert self.total_len // self.batch_size == \ assert self.total_len // self.batch_size == \
self.real_len // self.real_batch_size self.real_len // self.real_batch_size, f"Number of batches({self.total_len // self.batch_size}!={self.real_len // self.real_batch_size}) not match, total_len: {self.total_len}, batch_size: {self.batch_size}, real_len: {self.real_len}, real_batch_size: {self.real_batch_size}"
else: else:
self.real_len = self.total_len self.real_len = self.total_len
self.real_batch_size = self.batch_size self.real_batch_size = self.batch_size

2
python/jittor/models/alexnet.py
View File

@ -23,7 +23,7 @@ class AlexNet(nn.Module):
Example:: Example::
model = jittor.models.AlexNet(500) model = jittor.models.AlexNet(500)
x = jittor.random([10,224,224,3]) x = jittor.random([10,3,224,224])
y = model(x) # [10, 500] y = model(x) # [10, 500]
""" """