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() |