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

test_resnet changed

This commit is contained in:
Dun Liang 2021-01-17 22:44:57 +08:00
parent 3a438e21c8
commit 3734af07c0
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
python/jittor/test/test_resnet.py
View File

@ -116,9 +116,9 @@ class TestResnet(unittest.TestCase):
# Train Epoch: 0 [50/100 (50%)] Loss: 2.055014 Acc: 0.290000
if jt.in_mpi:
assert jt.core.number_of_lived_vars() < 7800, jt.core.number_of_lived_vars()
assert jt.core.number_of_lived_vars() < 8000, jt.core.number_of_lived_vars()
else:
assert jt.core.number_of_lived_vars() < 6700, jt.core.number_of_lived_vars()
assert jt.core.number_of_lived_vars() < 6900, jt.core.number_of_lived_vars()
jt.sync_all(True)
assert np.mean(loss_list[-50:])<0.5