Skip to content

GitLab

Unverified Commit 7d5d628e authored by ver217's avatar ver217 Committed by GitHub
Browse files

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

parent df544814
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_ddp/test_ddp_state_dict.py
View file @ 7d5d628e
...@@ -25,7 +25,7 @@ def check_state_dict_equal(state_dict: OrderedDict, other_state_dict: OrderedDic ...@@ -25,7 +25,7 @@ def check_state_dict_equal(state_dict: OrderedDict, other_state_dict: OrderedDic
else: else:
temp_t2 = t2 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: def init_ddp(module: torch.nn.Module) -> ColoDDP:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment