Skip to content

Commit

Permalink
fix precision for vit
Browse files Browse the repository at this point in the history
  • Loading branch information
Fridge003 committed Aug 9, 2023
1 parent 8613365 commit 98d83b5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/test_shardformer/test_model/test_shard_vit.py
Original file line number Diff line number Diff line change
Expand Up @@ -80,7 +80,7 @@ def check_forward_backward(model_fn, data_gen_fn, output_transform_fn, loss_fn,
shard_vit_model,
col_layer_for_check,
tp_group,
atol=1e-3,
atol=5e-3,
rtol=1e-3,
dim=1,
verbose=False)
Expand Down

0 comments on commit 98d83b5

Please sign in to comment.