mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-16 14:41:53 +00:00
[test] modify model supporting part of low_level_zero plugin (including correspoding docs)
This commit is contained in:
@@ -42,12 +42,6 @@ Zero-2 不支持局部梯度累积。如果您坚持使用,虽然可以积累
|
||||
|
||||
- `timm.models.convit_base`
|
||||
- dlrm and deepfm models in `torchrec`
|
||||
- `diffusers.VQModel`
|
||||
- `transformers.AlbertModel`
|
||||
- `transformers.AlbertForPreTraining`
|
||||
- `transformers.BertModel`
|
||||
- `transformers.BertForPreTraining`
|
||||
- `transformers.GPT2DoubleHeadsModel`
|
||||
|
||||
兼容性问题将在未来修复。
|
||||
|
||||
|
Reference in New Issue
Block a user