@ -18,7 +18,7 @@ logging.getLogger("matplotlib").setLevel(logging.ERROR)
MATPLOTLIB_FLAG = False
MATPLOTLIB_FLAG = False
logging . basicConfig ( stream = sys . stdout , level = logging . DEBU G)
logging . basicConfig ( stream = sys . stdout , level = logging . WARNIN G)
logger = logging
logger = logging
@ -310,13 +310,13 @@ def check_git_hash(model_dir):
def get_logger ( model_dir , filename = " train.log " ) :
def get_logger ( model_dir , filename = " train.log " ) :
global logger
global logger
logger = logging . getLogger ( os . path . basename ( model_dir ) )
logger = logging . getLogger ( os . path . basename ( model_dir ) )
logger . setLevel ( logging . DEBU G)
logger . setLevel ( logging . WARNIN G)
formatter = logging . Formatter ( " %(asctime)s \t %(name)s \t %(levelname)s \t %(message)s " )
formatter = logging . Formatter ( " %(asctime)s \t %(name)s \t %(levelname)s \t %(message)s " )
if not os . path . exists ( model_dir ) :
if not os . path . exists ( model_dir ) :
os . makedirs ( model_dir )
os . makedirs ( model_dir )
h = logging . FileHandler ( os . path . join ( model_dir , filename ) )
h = logging . FileHandler ( os . path . join ( model_dir , filename ) )
h . setLevel ( logging . DEBU G)
h . setLevel ( logging . WARNIN G)
h . setFormatter ( formatter )
h . setFormatter ( formatter )
logger . addHandler ( h )
logger . addHandler ( h )
return logger
return logger