[Fix] Fix stuck in torch.einsum when create dataset (#2065)

* Fix stuck in torch.einsum when create dataset * Fix stuck in torch.einsum when create dataset * Fix lint * add comments * Fix lint
open-mmlab · Dec 2, 2022 · 353985c · 353985c
1 parent 9abe207
commit 353985c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tools/create_data.py b/tools/create_data.py
@@ -256,6 +256,11 @@ def waymo_data_prep(root_path,
 if __name__ == '__main__':
     from mmdet3d.utils import register_all_modules
     register_all_modules()
+
+    # Set to spawn mode to avoid stuck when process dataset creating
+    import multiprocessing
+    multiprocessing.set_start_method('spawn')
+
     if args.dataset == 'kitti':
         kitti_data_prep(
             root_path=args.root_path,