init logger level and file handler
(debug=True, log_file=None)
| 548 | |
| 549 | |
| 550 | def init_logger(debug=True, log_file=None): |
| 551 | """ |
| 552 | init logger level and file handler |
| 553 | """ |
| 554 | if debug: |
| 555 | logger.setLevel(logging.DEBUG) |
| 556 | else: |
| 557 | logger.setLevel(logging.INFO) |
| 558 | |
| 559 | if log_file is not None: |
| 560 | logfHandler = logging.FileHandler(log_file) |
| 561 | logfHandler.setFormatter( |
| 562 | logging.Formatter( |
| 563 | "%(asctime)s - %(funcName)s:%(lineno)d - %(levelname)s - %(message)s" |
| 564 | ) |
| 565 | ) |
| 566 | logger.addHandler(logfHandler) |
| 567 | |
| 568 | |
| 569 | def check_test_mode(mode="cpu", gpu_id=0): |