ColossalAI/colossalai/nn
HELSON 2458659919
[zero] fix error for BEiT models (#2169)
* [zero] fix error for BEiT models

* [ColoParameter] add unpack operation for tuple arguments

* fix bugs

* fix chunkv2 unit testing

* add assertion for gradient state
2022-12-26 15:03:54 +08:00
..
_ops [zero] fix error for BEiT models (#2169) 2022-12-26 15:03:54 +08:00
layer [hotfix] Jit type hint #2161 (#2164) 2022-12-22 10:17:03 +08:00
loss
lr_scheduler
metric
optimizer [builder] unified cpu_optim fused_optim inferface (#2190) 2022-12-23 20:57:41 +08:00
parallel [zero] fix error for BEiT models (#2169) 2022-12-26 15:03:54 +08:00
__init__.py
init.py