github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[DDP] test ddp state dict uses more strict threshold (#1382)

pull/1384/head
ver217 2022-07-28 17:29:04 +08:00 committed by GitHub
parent df54481473
commit 7d5d628e07
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_ddp/test_ddp_state_dict.py
View File

@ -25,7 +25,7 @@ def check_state_dict_equal(state_dict: OrderedDict, other_state_dict: OrderedDic
else:
temp_t2 = t2
assert torch.allclose(t1, temp_t2, atol=1e-3, rtol=1e-3)
assert torch.equal(t1, temp_t2)
def init_ddp(module: torch.nn.Module) -> ColoDDP: