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)
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)