fastwam / scripts /train.py
jwfanDL's picture
Add files using upload-large-folder tool
7d0e316 verified
Raw
History Blame Contribute Delete
358 Bytes
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()