github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-03 01:55:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

[chore] remove unnecessary test & changes

Browse Source
This commit is contained in:
hxwang
2024-05-24 06:09:36 +00:00
parent ff507b755e
commit ca674549e0
7 changed files with 16 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_zero/test_gemini/test_grad_accum.py
View File

@@ -50,7 +50,7 @@ def check_grad(model: GeminiDDP, torch_model: torch.nn.Module):
@parameterize("model_name", ["transformers_gpt_lm"])
@parameterize("master_weights", [False, True])
@parameterize("use_grad_checkpoint", [False, True])
@parameterize("max_prefetch", [0, 1, 4])
@parameterize("max_prefetch", [0, 4])
@parameterize("enable_async_reduce", [False, True])
def exam_gemini_grad_acc(
placement_config,