diff --git a/parl/utils/logger.py b/parl/utils/logger.py
index 46815a417cd3a0f16a2298ca09193356855217eb..ac6f12d58ddc6919acdd8ededa391b23393ee3db 100644
--- 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)