From ad798302a2936920f7a7fec23de50128ea61803e Mon Sep 17 00:00:00 2001 From: ver217 Date: Fri, 21 Feb 2025 13:20:16 +0800 Subject: [PATCH] fix test --- tests/test_shardformer/test_model/test_shard_t5.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_shardformer/test_model/test_shard_t5.py b/tests/test_shardformer/test_model/test_shard_t5.py index 6cdf5bf41..67d4f7a7f 100644 --- a/tests/test_shardformer/test_model/test_shard_t5.py +++ b/tests/test_shardformer/test_model/test_shard_t5.py @@ -79,7 +79,7 @@ def check_forward_backward(model_fn, data_gen_fn, output_transform_fn, loss_fn, # TODO he precision in weight checking is too significant. atol, rtol = 1e-3, 1e-3 else: - atol, rtol = 5e-3, 5e-3 + atol, rtol = 6e-3, 0 if stage_manager is None or stage_manager.is_first_stage(): check_weight( t5,