Jiarui Fang 7650177713 [zero] global model data memory tracer (#360)

2022-03-10 11:20:04 +08:00

..

init_ctx

[zero] global model data memory tracer (#360)

2022-03-10 11:20:04 +08:00

shard_utils

[zero] Update sharded model v2 using sharded param v2 (#323)

2022-03-08 18:18:06 +08:00

sharded_model

[test] polish zero related unitest (#351)

2022-03-10 09:57:26 +08:00

sharded_optim

[zero] update sharded optim v2 (#334)

2022-03-09 16:09:36 +08:00

sharded_param

[zero] global model data memory tracer (#360)

2022-03-10 11:20:04 +08:00

__init__.py

added buffer sync to naive amp model wrapper (#291)

2022-03-02 16:47:17 +08:00