mirror of https://github.com/hpcaitech/ColossalAI
9880fd2cd8
* Fix state_dict output for ZeroDDP duplicated parameters * Rewrite state_dict based on get_static_torch_model * Modify get_static_torch_model to be compatible with the lower version (ZeroDDP) |
||
---|---|---|
.. | ||
layers | ||
__init__.py | ||
data_parallel.py | ||
gemini_parallel.py | ||
reducer.py | ||
utils.py |