fix test

2025-02-21 13:20:16 +08:00 · 2025-02-21 13:20:16 +08:00 · ad798302a2
parent 91d0d8fa85
commit ad798302a2
1 changed files with 1 additions and 1 deletions
--- 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,