pvnet/run.py at main · openclimatefix/pvnet · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
"""Run training.

This file can be run for example using
>>  python run.py experiment=example_simple
"""

import logging
import sys

import hydra
from omegaconf import DictConfig

from pvnet.training import train
from pvnet.utils import print_config, run_config_utilities, validate_gpu_config

logging.basicConfig(stream=sys.stdout, level=logging.ERROR)


@hydra.main(config_path="configs/", config_name="config.yaml", version_base="1.2")
def main(config: DictConfig) -> None:
    """Runs training"""

    # A couple of optional utilities:
    # - disabling python warnings
    # - forcing debug friendly configuration
    # - forcing multi-gpu friendly configuration
    run_config_utilities(config)
    validate_gpu_config(config)
    print_config(config, resolve=True)

    return train(config)


if __name__ == "__main__":
    main()