ColossalAI

Commit Graph

Author	SHA1	Message	Date
Frank Lee	3601b2bad0	[test] fixed rerun_on_exception and adapted test cases (#487 )	2022-03-25 17:25:12 +08:00
アマデウス	01a80cd86d	Hotfix/Colossalai layers (#92 ) * optimized 1d layer apis; reorganized nn.layer modules; fixed tests * fixed 2.5d runtime issue * reworked split batch, now called in trainer.schedule.load_batch Co-authored-by: BoxiangW <45734921+BoxiangW@users.noreply.github.com>	2021-12-29 23:32:10 +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

Author

SHA1

Message

Date

Frank Lee

3601b2bad0

[test] fixed rerun_on_exception and adapted test cases (#487 )

2022-03-25 17:25:12 +08:00

アマデウス

01a80cd86d

Hotfix/Colossalai layers (#92 )

* optimized 1d layer apis; reorganized nn.layer modules; fixed tests

* fixed 2.5d runtime issue

* reworked split batch, now called in trainer.schedule.load_batch

Co-authored-by: BoxiangW <45734921+BoxiangW@users.noreply.github.com>

2021-12-29 23:32:10 +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

3 Commits (20ab1f55204ee18aca204f6faf60ca27b509b509)