File size: 342 Bytes
b4efe93 | 1 2 3 4 5 6 7 8 9 | from hydra.core.config_store import ConfigStore
from trainer.optimizers.adamw import AdamWOptimizerConfig
from trainer.optimizers.dummy_optimizer import DummyOptimizerConfig
cs = ConfigStore.instance()
cs.store(group="optimizer", name="dummy", node=DummyOptimizerConfig)
cs.store(group="optimizer", name="adamw", node=AdamWOptimizerConfig)
|