mirror of https://github.com/hpcaitech/ColossalAI
2d1a7dfe5f
* [zero] add strict ddp mode * [polish] add comments for strict ddp mode * [zero] fix test error |
||
---|---|---|
.. | ||
common_utils | ||
core | ||
model | ||
test_colo_checkpoint_tools.py | ||
test_comm_spec_apply.py | ||
test_context.py | ||
test_mix_gather.py | ||
test_parameter.py | ||
test_shape_consistency.py | ||
test_shape_consistency_apply.py | ||
test_sharded_linear.py | ||
test_sharding_spec.py | ||
test_tp_with_zero.py |