github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-05 12:24:38 +00:00
Code Issues Packages Projects Releases Wiki Activity
39 Commits 62 Branches 51 Tags
2c0c85d3d337c43748ddd3ea23dfd2417de79f11
Commit Graph

3 Commits

Author SHA1 Message Date
ver217
f03bcb359b update vit example for new API (#98) (#99) 2022-01-04 20:35:33 +08:00
アマデウス
0fedef4f3c Layer integration (#83) 
* integrated parallel layers for ease of building models

* integrated 2.5d layers

* cleaned codes and unit tests

* added log metric by step hook; updated imagenet benchmark; fixed some bugs

* reworked initialization; cleaned codes

Co-authored-by: BoxiangW <45734921+BoxiangW@users.noreply.github.com>
 2021-12-27 15:04:32 +08:00
Frank Lee
35813ed3c4 update examples and sphnix docs for the new api (#63) 2021-12-13 22:07:01 +08:00