Segmentation fault (core dumped)

[yeyang1021]

The enviroment:
Pytorch 1.10.1
CUDA 11.1
GPU: gtx 3090
When I use python train.py --cfg_file cfgs/kitti_models/voxset.yaml.

Segmentation fault (core dumped)

cfg.MODEL.POST_PROCESSING.NMS_CONFIG = edict()
2022-04-18 11:11:59,579 INFO cfg.MODEL.POST_PROCESSING.NMS_CONFIG.MULTI_CLASSES_NMS: False
2022-04-18 11:11:59,580 INFO cfg.MODEL.POST_PROCESSING.NMS_CONFIG.NMS_TYPE: nms_gpu
2022-04-18 11:11:59,580 INFO cfg.MODEL.POST_PROCESSING.NMS_CONFIG.NMS_THRESH: 0.01
2022-04-18 11:11:59,580 INFO cfg.MODEL.POST_PROCESSING.NMS_CONFIG.NMS_PRE_MAXSIZE: 4096
2022-04-18 11:11:59,580 INFO cfg.MODEL.POST_PROCESSING.NMS_CONFIG.NMS_POST_MAXSIZE: 500
2022-04-18 11:11:59,580 INFO
cfg.OPTIMIZATION = edict()
2022-04-18 11:11:59,580 INFO cfg.OPTIMIZATION.BATCH_SIZE_PER_GPU: 4
2022-04-18 11:11:59,580 INFO cfg.OPTIMIZATION.NUM_EPOCHS: 100
2022-04-18 11:11:59,580 INFO cfg.OPTIMIZATION.OPTIMIZER: adam_onecycle
2022-04-18 11:11:59,580 INFO cfg.OPTIMIZATION.LR: 0.0015
2022-04-18 11:11:59,580 INFO cfg.OPTIMIZATION.WEIGHT_DECAY: 0.01
2022-04-18 11:11:59,580 INFO cfg.OPTIMIZATION.MOMENTUM: 0.9
2022-04-18 11:11:59,580 INFO cfg.OPTIMIZATION.MOMS: [0.95, 0.85]
2022-04-18 11:11:59,580 INFO cfg.OPTIMIZATION.PCT_START: 0.4
2022-04-18 11:11:59,580 INFO cfg.OPTIMIZATION.DIV_FACTOR: 10
2022-04-18 11:11:59,580 INFO cfg.OPTIMIZATION.DECAY_STEP_LIST: [35, 45]
2022-04-18 11:11:59,580 INFO cfg.OPTIMIZATION.LR_DECAY: 0.1
2022-04-18 11:11:59,580 INFO cfg.OPTIMIZATION.LR_CLIP: 1e-07
2022-04-18 11:11:59,580 INFO cfg.OPTIMIZATION.LR_WARMUP: False
2022-04-18 11:11:59,580 INFO cfg.OPTIMIZATION.WARMUP_EPOCH: 1
2022-04-18 11:11:59,580 INFO cfg.OPTIMIZATION.GRAD_NORM_CLIP: 10
2022-04-18 11:11:59,580 INFO cfg.TAG: voxset
2022-04-18 11:11:59,580 INFO cfg.EXP_GROUP_PATH: kitti_models
2022-04-18 11:11:59,776 INFO Database filter by min points Car: 14357 => 13532
2022-04-18 11:11:59,777 INFO Database filter by min points Pedestrian: 2207 => 2168
2022-04-18 11:11:59,777 INFO Database filter by min points Cyclist: 734 => 705
2022-04-18 11:11:59,797 INFO Database filter by difficulty Car: 13532 => 10759
2022-04-18 11:11:59,801 INFO Database filter by difficulty Pedestrian: 2168 => 2075
2022-04-18 11:11:59,802 INFO Database filter by difficulty Cyclist: 705 => 581
2022-04-18 11:11:59,809 INFO Loading KITTI dataset
2022-04-18 11:11:59,949 INFO Total samples for KITTI dataset: 3712
Segmentation fault (core dumped)

[OrangeSodahub]

@yeyang1021 Hello, have you solved this problem?