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