colossalai/utils/checkpoint/utils.py · 7a8702c06dbcde8e01102dc0d4804740bc5dafcd · OpenDAS / ColossalAI · GitLab

Find file Blame History Permalink

[colotensor] add Tensor.view op and its unit test (#1343) · 7a8702c0
HELSON authored Jul 21, 2022
```
[colotensor] add megatron initialization for gpt2
```
7a8702c0

utils.py 1.83 KB