import os import sys import time import logging from datetime import datetime def init_log(output_dir): logging.basicConfig(level=logging.DEBUG, format='%(asctime)s %(message)s', datefmt='%Y%m%d-%H:%M:%S', filename=os.path.join(output_dir, datetime.now().strftime('%Y%m%d_%H%M%S')+'.log'), filemode='w') console = logging.StreamHandler() console.setLevel(logging.INFO) logging.getLogger('').addHandler(console) return logging