github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-16 06:30:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
ver217
2022-07-28 17:29:04 +08:00
committed by GitHub
parent df54481473
commit 7d5d628e07
1 changed files with 1 additions and 1 deletions
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: