aryadomain's picture
Add files using upload-large-folder tool
ef8f3ad verified
Raw
History Blame Contribute Delete
342 Bytes
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)