ColossalAI/colossalai/nn
Edenzzzz 15055f9a36
[hotfix] quick fixes to make legacy tutorials runnable (#5559)
Co-authored-by: Edenzzzz <wtan45@wisc.edu>
2024-04-07 12:06:27 +08:00
..
layer [hotfix] quick fixes to make legacy tutorials runnable (#5559) 2024-04-07 12:06:27 +08:00
loss [moe] merge moe into main (#4978) 2023-11-02 02:21:24 +00:00
lr_scheduler [lr-scheduler] fix load state dict and add test (#5369) 2024-02-06 14:23:32 +08:00
optimizer [hotfix] fix typo s/keywrods/keywords etc. (#5429) 2024-03-12 11:25:16 +08:00
__init__.py
init.py