forked from kredde/compression-robustness
-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
28 lines (18 loc) · 618 Bytes
/
main.py
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
import logging
import dotenv
import hydra
from omegaconf import DictConfig
dotenv.load_dotenv(override=True)
logger = logging.getLogger(__name__)
@hydra.main(config_path="configs/", config_name="config.yaml")
def main(config: DictConfig):
logger.info(config.pretty())
# Imports should be nested inside @hydra.main to optimize tab completion
# Read more here: https://github.com/facebookresearch/hydra/issues/934
from src.train import train
from src.utils import config_utils
config_utils.extras(config)
# Train model
return train(config)
if __name__ == "__main__":
main()