From 7d5d628e07915a81b2d9ecaf74f92be3b59b497b Mon Sep 17 00:00:00 2001 From: ver217 Date: Thu, 28 Jul 2022 17:29:04 +0800 Subject: [PATCH] [DDP] test ddp state dict uses more strict threshold (#1382) --- tests/test_ddp/test_ddp_state_dict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_ddp/test_ddp_state_dict.py b/tests/test_ddp/test_ddp_state_dict.py index 3f3c316a8..359dcafac 100644 --- a/tests/test_ddp/test_ddp_state_dict.py +++ b/tests/test_ddp/test_ddp_state_dict.py @@ -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: