File size: 358 Bytes
7d0e316 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | import hydra
from omegaconf import DictConfig
from fastwam.runtime import run_training
from fastwam.utils.config_resolvers import register_default_resolvers
register_default_resolvers()
@hydra.main(config_path="../configs", config_name="train", version_base="1.3")
def main(cfg: DictConfig):
run_training(cfg)
if __name__ == "__main__":
main()
|