Main entry function for any of the tasks.
Source code in quadra/main.py
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41 | @hydra.main(config_path="configs/", config_name="config.yaml", version_base="1.3.0")
def main(config: DictConfig):
"""Main entry function for any of the tasks."""
if config.validate:
start = time.time()
validate_config(config)
stop = time.time()
log.info("Config validation took %f seconds", stop - start)
from quadra.utils import utils # pylint: disable=import-outside-toplevel
utils.extras(config)
# Prints the resolved configuration to the console
if config.get("print_config"):
utils.print_config(config, resolve=True)
# Set seed for random number generators in pytorch, numpy and python.random
seed_everything(config.core.seed, workers=True)
setup_opencv()
# Run specified task using the configuration composition
task: Task = hydra.utils.instantiate(config.task, config, _recursive_=False)
task.execute()
|