| import hydra | |
| from omegaconf import DictConfig, OmegaConf | |
| from src.train import train | |
| @hydra.main(version_base="1.3", config_path=".", config_name="train.yaml") | |
| def main(cfg: DictConfig) -> None: | |
| print(OmegaConf.to_yaml(cfg)) | |
| train(cfg) | |
| if __name__ == "__main__": | |
| main() |