| from hydra import compose, initialize_config_dir | |
| from omegaconf import OmegaConf | |
| from src.train import train | |
| def main() -> None: | |
| with initialize_config_dir(version_base="1.3", config_dir="."): | |
| cfg = compose( | |
| config_name="train.yaml", return_hydra_config=True, overrides=[] | |
| ) | |
| print(OmegaConf.to_yaml(cfg)) | |
| train(cfg) | |
| if __name__ == "__main__": | |
| main() |