mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-10-22 15:26:57 +00:00
569357fea0
* add pytorch hooks fix #175 * remove licenses in src code * add gpu memory tracer * replacing print with logger in ophooks.
13 lines
663 B
Python
13 lines
663 B
Python
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, build_ophooks)
|
|
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', 'build_ophooks'
|
|
]
|