fix auto set dir (#360)

0915559a · Zheyue Tan · GitHub · a7ac16df · 0915559a
隐藏空白更改
内联并排

Showing with 4 addition and 1 deletion

parl/utils/logger.py parl/utils/logger.py +4 -1

未找到文件。
--- a/parl/utils/logger.py
+++ b/parl/utils/logger.py
@@ -170,7 +170,10 @@ def auto_set_dir(action=None):
        dirname(str): log directory used in the global logging directory.
    """
    mod = sys.modules['__main__']
-    basename = os.path.basename(mod.__file__)
+    if hasattr(mod, '__file__'):
+        basename = os.path.basename(mod.__file__)
+    else:
+        basename = ''
    dirname = os.path.join('train_log', basename[:basename.rfind('.')])
    dirname = os.path.normpath(dirname)