diff --git a/inference.py b/inference.py index 05defb2..2e68373 100644 --- a/inference.py +++ b/inference.py @@ -13,6 +13,14 @@ from tools import * +# Configure logging +logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s') +logger = logging.getLogger(__name__) + if __name__ == '__main__': + logger.info("Loading configuration") cfg_update = Config(load=True) + logger.info("Building inference engine") INFER_ENGINE.build(dict(type=cfg_update.TASK_TYPE), cfg_update=cfg_update.cfg_dict) + logger.info("Inference engine built successfully") +