File size: 446 Bytes
533920b | 1 2 3 4 5 6 7 8 9 10 11 | from hydra.core.config_store import ConfigStore
from trainer.accelerators.debug_accelerator import DebugAcceleratorConfig
from trainer.accelerators.deepspeed_accelerator import DeepSpeedAcceleratorConfig
ACCELERATOR_GROUP_NAME = "accelerator"
cs = ConfigStore.instance()
cs.store(group=ACCELERATOR_GROUP_NAME, name="deepspeed", node=DeepSpeedAcceleratorConfig)
cs.store(group=ACCELERATOR_GROUP_NAME, name="debug", node=DebugAcceleratorConfig)
|