Can release code for train teacher model?

Cc-Hy / CMKD

Cross-Modality Knowledge Distillation Network for Monocular 3D Object Detection (ECCV 2022 Oral)

Apache License 2.0

108 stars 9 forks source link

Can release code for train teacher model? #3

Closed xiaowenhe closed 1 year ago

xiaowenhe commented 1 year ago

Hi, Can release code for train teacher model? Thank you !

Cc-Hy commented 1 year ago

Hi, this repo is compatible with OpenPCDet, so you can follow the official instruction to train any LiDAR-based teacher model.

To train the teacher model in the paper, use CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 train.py --launcher pytorch --cfg cfgs/kitti_models/second_teacher.yaml