main
This commit is contained in:
parent
74d8e2602c
commit
674bd0d0e2
|
|
@ -0,0 +1,19 @@
|
||||||
|
import hydra
|
||||||
|
import torch
|
||||||
|
from hydra.core.config_store import ConfigStore
|
||||||
|
|
||||||
|
from enhancer.utils.config import EnhancerConfig
|
||||||
|
|
||||||
|
cs = ConfigStore.instance()
|
||||||
|
cs.store(name="enhancer_config", node=EnhancerConfig)
|
||||||
|
|
||||||
|
|
||||||
|
@hydra.main(config_path=".",config_name="conf")
|
||||||
|
def main(cfg: EnhancerConfig):
|
||||||
|
|
||||||
|
print(cfg.paths.data)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
if __name__=="__main__":
|
||||||
|
main()
|
||||||
Loading…
Reference in New Issue