utils/log.py

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