| 106 | import torch |
| 107 | |
| 108 | def configure_logging(): |
| 109 | logger = logging.getLogger("sat") |
| 110 | logger.setLevel(os.environ.get("SAT_LOGLEVEL", "INFO")) |
| 111 | if os.environ.get("LOGLEVEL", None) is not None: |
| 112 | logger.setLevel(os.environ.get("LOGLEVEL", "INFO")) |
| 113 | formatter = logging.Formatter('[%(asctime)s] [%(levelname)s] %(message)s') |
| 114 | |
| 115 | # stream handler |
| 116 | sh = logging.StreamHandler() |
| 117 | logger.setLevel(os.environ.get("SAT_LOGLEVEL", "INFO")) |
| 118 | if os.environ.get("LOGLEVEL", None) is not None: |
| 119 | logger.setLevel(os.environ.get("LOGLEVEL", "INFO")) |
| 120 | sh.setFormatter(formatter) |
| 121 | logger.addHandler(sh) |
| 122 | |
| 123 | return logger |
| 124 | |
| 125 | logger = configure_logging() |
| 126 | |