mirror of https://github.com/hpcaitech/ColossalAI
![]() * [zero] fix error for BEiT models * [ColoParameter] add unpack operation for tuple arguments * fix bugs * fix chunkv2 unit testing * add assertion for gradient state |
||
---|---|---|
.. | ||
update | ||
test_gemini_manager.py | ||
test_param_op.py | ||
test_runtime_mem_tracer.py |