store

BaseLitModule Hydra config store.

cneuromax.fitting.deeplearning.litmodule.store.store_basic_nnmodule_config(store)

Stores hydra litmodule/nnmodule group config.

Ref: Hydra

Config name: mlp.

Parameters:

store (ZenStore) – See store.

Return type:

None

cneuromax.fitting.deeplearning.litmodule.store.store_basic_optimizer_configs(store)

Stores hydra litmodule/optimizer group configs.

Ref: Hydra.

Config names: adam, adamw, sgd.

Parameters:

store (ZenStore) – See store.

Return type:

None

cneuromax.fitting.deeplearning.litmodule.store.store_basic_scheduler_configs(store)

Stores hydra litmodule/scheduler group configs.

Ref: Hydra.

Config names: constant, linear_warmup.

Parameters:

store (ZenStore) – See store.

Return type:

None