提交 3c46b7c4 编写于 作者: H Hongsheng Zeng 提交者: Bo Zhou

fix bug of logger (#29)

上级 e11b40c5
......@@ -134,7 +134,8 @@ def get_dir():
# Will save log to log_dir/main_file_name/log.log by default
mod = sys.modules['__main__']
basename = os.path.basename(mod.__file__)
auto_dirname = os.path.join('log_dir', basename[:basename.rfind('.')])
set_dir(auto_dirname)
_logger.info("Argv: " + ' '.join(sys.argv))
if hasattr(mod, '__file__'):
basename = os.path.basename(mod.__file__)
auto_dirname = os.path.join('log_dir', basename[:basename.rfind('.')])
set_dir(auto_dirname)
_logger.info("Argv: " + ' '.join(sys.argv))
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册