mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-11 05:49:55 +00:00
@@ -9,6 +9,7 @@ colossalai.nn.optimizer
|
||||
:maxdepth: 2
|
||||
|
||||
colossalai.nn.optimizer.colossalai_optimizer
|
||||
colossalai.nn.optimizer.cpu_adam
|
||||
colossalai.nn.optimizer.fused_adam
|
||||
colossalai.nn.optimizer.fused_lamb
|
||||
colossalai.nn.optimizer.fused_sgd
|
||||
|
Reference in New Issue
Block a user