github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-16 22:52:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

[hotfix]: add pp sanity check and fix mbs arg (#5268)

Browse Source 
* fix: fix misleading mbs arg

* feat: add pp sanity check

* fix: fix 1f1b sanity check
This commit is contained in:
Wenhao Chen
2024-01-15 15:57:40 +08:00
committed by GitHub
parent c174c4fc5f
commit ef4f0ee854
5 changed files with 21 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_pipeline/test_schedule/test_oneF_oneB.py
View File

@@ -155,7 +155,7 @@ def run_dist(
@pytest.mark.dist
@pytest.mark.parametrize("num_microbatch", [4, 12])
@pytest.mark.parametrize("num_microbatch", [4, 6])
@pytest.mark.parametrize("batch_size", [12])
@pytest.mark.parametrize("world_size", [2, 4])
@rerun_if_address_is_in_use()