fixed the venv issue

Lifelong-Robot-Learning · Dec 25, 2023 · f78abd6 · f78abd6
1 parent a73c497
commit f78abd6
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/libero/lifelong/main.py b/libero/lifelong/main.py
@@ -3,6 +3,7 @@
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 import sys
 import json
+import multiprocessing
 import pprint
 import time
 from pathlib import Path
@@ -265,4 +266,7 @@ def main(hydra_cfg):
 
 
 if __name__ == "__main__":
+    # Set the multiprocessing start method to 'spawn'
+    if multiprocessing.get_start_method(allow_none=True) != "spawn":  
+        multiprocessing.set_start_method("spawn", force=True)
     main()