main.py

import hydra
from omegaconf import DictConfig, OmegaConf
from train import train


@hydra.main(version_base=None, config_path="conf", config_name="config")
def main(cfg: DictConfig):
    print(OmegaConf.to_yaml(cfg))
    train(cfg)


if __name__ == '__main__':
    main()