.. |
colossalai.amp.apex_amp.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.amp.naive_amp.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.amp.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.amp.torch_amp.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.builder.builder.rst
|
Migrated project
|
3 years ago |
colossalai.builder.pipeline.rst
|
Migrated project
|
3 years ago |
colossalai.builder.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.communication.collective.rst
|
Migrated project
|
3 years ago |
colossalai.communication.p2p.rst
|
Migrated project
|
3 years ago |
colossalai.communication.ring.rst
|
Migrated project
|
3 years ago |
colossalai.communication.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.communication.utils.rst
|
Migrated project
|
3 years ago |
colossalai.constants.rst
|
Migrated project
|
3 years ago |
colossalai.context.config.rst
|
Migrated project
|
3 years ago |
colossalai.context.parallel_context.rst
|
Migrated project
|
3 years ago |
colossalai.context.parallel_mode.rst
|
Migrated project
|
3 years ago |
colossalai.context.process_group_initializer.initializer_1d.rst
|
Migrated project
|
3 years ago |
colossalai.context.process_group_initializer.initializer_2d.rst
|
Migrated project
|
3 years ago |
colossalai.context.process_group_initializer.initializer_2p5d.rst
|
Migrated project
|
3 years ago |
colossalai.context.process_group_initializer.initializer_3d.rst
|
Migrated project
|
3 years ago |
colossalai.context.process_group_initializer.initializer_data.rst
|
Migrated project
|
3 years ago |
colossalai.context.process_group_initializer.initializer_pipeline.rst
|
Migrated project
|
3 years ago |
colossalai.context.process_group_initializer.initializer_sequence.rst
|
Migrated project
|
3 years ago |
colossalai.context.process_group_initializer.initializer_tensor.rst
|
Migrated project
|
3 years ago |
colossalai.context.process_group_initializer.process_group_initializer.rst
|
Migrated project
|
3 years ago |
colossalai.context.process_group_initializer.rst
|
Migrated project
|
3 years ago |
colossalai.context.random.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.context.random.seed_manager.rst
|
Migrated project
|
3 years ago |
colossalai.context.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.core.rst
|
Migrated project
|
3 years ago |
colossalai.engine.gradient_handler.rst
|
Migrated project
|
3 years ago |
colossalai.engine.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.engine.schedule.rst
|
Migrated project
|
3 years ago |
colossalai.initialize.rst
|
Migrated project
|
3 years ago |
colossalai.logging.logging.rst
|
Migrated project
|
3 years ago |
colossalai.logging.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.nn.layer.base_layer.rst
|
Migrated project
|
3 years ago |
colossalai.nn.layer.non_parallel_layers.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.nn.layer.parallel_1d.layers.rst
|
Migrated project
|
3 years ago |
colossalai.nn.layer.parallel_1d.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.nn.layer.parallel_2d.layers.rst
|
Migrated project
|
3 years ago |
colossalai.nn.layer.parallel_2d.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.nn.layer.parallel_2p5d.layers.rst
|
Migrated project
|
3 years ago |
colossalai.nn.layer.parallel_2p5d.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.nn.layer.parallel_3d.layers.rst
|
Migrated project
|
3 years ago |
colossalai.nn.layer.parallel_3d.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.nn.layer.parallel_sequence.layers.rst
|
Migrated project
|
3 years ago |
colossalai.nn.layer.parallel_sequence.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.nn.layer.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.nn.layer.wrapper.lambda_wrapper.rst
|
Migrated project
|
3 years ago |
colossalai.nn.layer.wrapper.rst
|
Migrated project
|
3 years ago |
colossalai.nn.loss.cross_entropy_2d.rst
|
Migrated project
|
3 years ago |
colossalai.nn.loss.cross_entropy_2p5d.rst
|
Migrated project
|
3 years ago |
colossalai.nn.loss.cross_entropy_3d.rst
|
Migrated project
|
3 years ago |
colossalai.nn.loss.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.nn.lr_scheduler.cosine.rst
|
Migrated project
|
3 years ago |
colossalai.nn.lr_scheduler.delayed.rst
|
Migrated project
|
3 years ago |
colossalai.nn.lr_scheduler.linear.rst
|
Migrated project
|
3 years ago |
colossalai.nn.lr_scheduler.multistep.rst
|
Migrated project
|
3 years ago |
colossalai.nn.lr_scheduler.onecycle.rst
|
Migrated project
|
3 years ago |
colossalai.nn.lr_scheduler.poly.rst
|
Migrated project
|
3 years ago |
colossalai.nn.lr_scheduler.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.nn.lr_scheduler.torch.rst
|
Migrated project
|
3 years ago |
colossalai.nn.model.model_from_config.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.nn.model.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.nn.optimizer.fused_adam.rst
|
Migrated project
|
3 years ago |
colossalai.nn.optimizer.fused_lamb.rst
|
Migrated project
|
3 years ago |
colossalai.nn.optimizer.fused_sgd.rst
|
Migrated project
|
3 years ago |
colossalai.nn.optimizer.lamb.rst
|
Migrated project
|
3 years ago |
colossalai.nn.optimizer.lars.rst
|
update documentation
|
3 years ago |
colossalai.nn.optimizer.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.nn.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.registry.registry.rst
|
Migrated project
|
3 years ago |
colossalai.registry.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.trainer.hooks.rst
|
Migrated project
|
3 years ago |
colossalai.trainer.metric.rst
|
Migrated project
|
3 years ago |
colossalai.trainer.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.utils.activation_checkpoint.rst
|
Migrated project
|
3 years ago |
colossalai.utils.checkpointing.rst
|
Support TP-compatible Torch AMP and Update trainer API (#27)
|
3 years ago |
colossalai.utils.common.rst
|
Migrated project
|
3 years ago |
colossalai.utils.cuda.rst
|
Migrated project
|
3 years ago |
colossalai.utils.data_sampler.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.utils.gradient_accumulation.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.utils.memory.rst
|
Migrated project
|
3 years ago |
colossalai.utils.multi_tensor_apply.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.utils.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |
colossalai.utils.timer.rst
|
Migrated project
|
3 years ago |
colossalai.zero.rst
|
update examples and sphnix docs for the new api (#63)
|
3 years ago |