Hongxin Liu
|
12eff9eb4c
|
[gemini] state dict supports fp16 (#3590)
* [gemini] save state dict support fp16
* [gemini] save state dict shard support fp16
* [gemini] fix state dict
* [gemini] fix state dict
|
2023-04-19 11:01:48 +08:00 |
|
Hongxin Liu
|
f313babd11
|
[gemini] support save state dict in shards (#3581)
* [gemini] support state dict shard
* [gemini] add test state dict shard
* [gemini] polish docstr
* [gemini] fix merge
* [gemini] polish code
|
2023-04-17 17:11:09 +08:00 |
|
YH
|
d329c294ec
|
Add docstr for zero3 chunk search utils (#3572)
|
2023-04-17 12:44:17 +08:00 |
|
Hongxin Liu
|
173dad0562
|
[misc] add verbose arg for zero and op builder (#3552)
* [misc] add print verbose
* [gemini] add print verbose
* [zero] add print verbose for low level
* [misc] add print verbose for op builder
|
2023-04-17 11:25:35 +08:00 |
|
Hongxin Liu
|
152239bbfa
|
[gemini] gemini supports lazy init (#3379)
* [gemini] fix nvme optimizer init
* [gemini] gemini supports lazy init
* [gemini] add init example
* [gemini] add fool model
* [zero] update gemini ddp
* [zero] update init example
* add chunk method
* add chunk method
* [lazyinit] fix lazy tensor tolist
* [gemini] fix buffer materialization
* [misc] remove useless file
* [booster] update gemini plugin
* [test] update gemini plugin test
* [test] fix gemini plugin test
* [gemini] fix import
* [gemini] fix import
* [lazyinit] use new metatensor
* [lazyinit] use new metatensor
* [lazyinit] fix __set__ method
|
2023-04-12 16:03:25 +08:00 |
|
YH
|
bcf0cbcbe7
|
[doc] Add docs for clip args in zero optim (#3504)
|
2023-04-10 11:11:28 +08:00 |
|
ver217
|
26b7aac0be
|
[zero] reorganize zero/gemini folder structure (#3424)
* [zero] refactor low-level zero folder structure
* [zero] fix legacy zero import path
* [zero] fix legacy zero import path
* [zero] remove useless import
* [zero] refactor gemini folder structure
* [zero] refactor gemini folder structure
* [zero] refactor legacy zero import path
* [zero] refactor gemini folder structure
* [zero] refactor gemini folder structure
* [zero] refactor gemini folder structure
* [zero] refactor legacy zero import path
* [zero] fix test import path
* [zero] fix test
* [zero] fix circular import
* [zero] update import
|
2023-04-04 13:48:16 +08:00 |