mirror of https://github.com/hpcaitech/ColossalAI
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
11 lines
605 B
11 lines
605 B
from .builder import (build_schedule, build_lr_scheduler, build_model, build_optimizer, build_layer,
|
|
build_loss, build_hooks, build_dataset, build_transform, build_data_sampler,
|
|
build_gradient_handler)
|
|
from .pipeline import build_pipeline_model, build_pipeline_model_from_cfg
|
|
|
|
__all__ = [
|
|
'build_schedule', 'build_lr_scheduler', 'build_model', 'build_optimizer',
|
|
'build_layer', 'build_loss', 'build_hooks', 'build_dataset', 'build_transform', 'build_data_sampler',
|
|
'build_gradient_handler', 'build_pipeline_model', 'build_pipeline_model_from_cfg'
|
|
]
|