Skip to content

Commit cfb6b33

Browse files
PeterH0323ZwwWayne
authored andcommitted
[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
1 parent cb7c679 commit cfb6b33

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tools/create_data.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -256,6 +256,11 @@ def waymo_data_prep(root_path,
256256
if __name__ == '__main__':
257257
from mmdet3d.utils import register_all_modules
258258
register_all_modules()
259+
260+
# Set to spawn mode to avoid stuck when process dataset creating
261+
import multiprocessing
262+
multiprocessing.set_start_method('spawn')
263+
259264
if args.dataset == 'kitti':
260265
kitti_data_prep(
261266
root_path=args.root_path,

0 commit comments

Comments
 (0)