RuntimeError: CUDA error: out of memory

[BigPig117]

hello, this is my error when i try to train the centerpoint.yaml

cfg.OPTIMIZATION = edict()
2022-04-07 11:02:52,498 INFO cfg.OPTIMIZATION.BATCH_SIZE_PER_GPU: 4
2022-04-07 11:02:52,498 INFO cfg.OPTIMIZATION.NUM_EPOCHS: 80
2022-04-07 11:02:52,498 INFO cfg.OPTIMIZATION.OPTIMIZER: adam_onecycle
2022-04-07 11:02:52,498 INFO cfg.OPTIMIZATION.LR: 0.003
2022-04-07 11:02:52,498 INFO cfg.OPTIMIZATION.WEIGHT_DECAY: 0.01
2022-04-07 11:02:52,498 INFO cfg.OPTIMIZATION.MOMENTUM: 0.9
2022-04-07 11:02:52,498 INFO cfg.OPTIMIZATION.MOMS: [0.95, 0.85]
2022-04-07 11:02:52,498 INFO cfg.OPTIMIZATION.PCT_START: 0.4
2022-04-07 11:02:52,498 INFO cfg.OPTIMIZATION.DIV_FACTOR: 10
2022-04-07 11:02:52,498 INFO cfg.OPTIMIZATION.DECAY_STEP_LIST: [35, 45]
2022-04-07 11:02:52,498 INFO cfg.OPTIMIZATION.LR_DECAY: 0.1
2022-04-07 11:02:52,498 INFO cfg.OPTIMIZATION.LR_CLIP: 1e-07
2022-04-07 11:02:52,498 INFO cfg.OPTIMIZATION.LR_WARMUP: False
2022-04-07 11:02:52,498 INFO cfg.OPTIMIZATION.WARMUP_EPOCH: 1
2022-04-07 11:02:52,498 INFO cfg.OPTIMIZATION.GRAD_NORM_CLIP: 10
2022-04-07 11:02:52,498 INFO cfg.TAG: centerpoint
2022-04-07 11:02:52,498 INFO cfg.EXP_GROUP_PATH: kitti_models
2022-04-07 11:02:52,570 INFO Database filter by min points Car: 14357 => 13532
2022-04-07 11:02:52,571 INFO Database filter by min points Pedestrian: 2207 => 2168
2022-04-07 11:02:52,571 INFO Database filter by min points Cyclist: 734 => 705
2022-04-07 11:02:52,582 INFO Database filter by difficulty Car: 13532 => 10759
2022-04-07 11:02:52,584 INFO Database filter by difficulty Pedestrian: 2168 => 2075
2022-04-07 11:02:52,585 INFO Database filter by difficulty Cyclist: 705 => 581
2022-04-07 11:02:52,588 INFO Loading KITTI dataset
2022-04-07 11:02:52,646 INFO Total samples for KITTI dataset: 3712
Traceback (most recent call last):
File "train.py", line 202, in
main()
File "train.py", line 116, in main
model = build_network(model_cfg=cfg.MODEL, num_class=len(cfg.CLASS_NAMES), dataset=train_set)
File "../pcdet/models/init.py", line 18, in build_network
model_cfg=model_cfg, num_class=num_class, dataset=dataset
File "../pcdet/models/detectors/init.py", line 30, in build_detector
model_cfg=model_cfg, num_class=num_class, dataset=dataset
File "../pcdet/models/detectors/centerpoint.py", line 7, in init
self.module_list = self.build_networks()
File "../pcdet/models/detectors/detector3d_template.py", line 47, in build_networks
model_info_dict=model_info_dict
File "../pcdet/models/detectors/detector3d_template.py", line 136, in build_dense_head
voxel_size=model_info_dict.get('voxel_size', False)
File "../pcdet/models/dense_heads/center_head.py", line 66, in init
[self.class_names.index(x) for x in cur_class_names if x in class_names]
RuntimeError: CUDA error: out of memory

base on GTX3070 cuda11.1 pytorch1.8.2 , can u give me some suggestions