RunpeiDong / PointDistiller

[CVPR 2023] PointDistiller: Structured Knowledge Distillation Towards Efficient and Compact 3D Detection
https://arxiv.org/abs/2205.11098
MIT License
66 stars 1 forks source link

Problem about Reimplementation!!! #12

Open lubinBoooos opened 2 weeks ago

lubinBoooos commented 2 weeks ago

Dear author, I try ur code to reimplement the performance of KD, but I cannot get same result as ur paper mentioned. Can u provide some training log about the best performance?

RunpeiDong commented 2 weeks ago

Hi @lubinBoooos,

It's here.

lubinBoooos commented 2 weeks ago

Hi @RunpeiDong, I followed ur log, while training, I found that my KD loss decreasing faster, and I cannot find the kd_cfg setting in ur log, so I want to know ur hyperparameters: num_voxels and kneighbours settings. Currently, I set it by num_voxels=6000, kneighbours=128.

RunpeiDong commented 2 weeks ago

Which model and which dataset?

lubinBoooos commented 2 weeks ago

pointpillar 16x on KITTI dataset, I got result: Overall AP40@easy, moderate, hard: bbox AP40:72.5636, 64.4526, 61.3323 bev AP40:70.5804, 61.2102, 57.6306 3d AP40:63.7857, 52.1458, 48.4905 aos AP40:61.45, 53.31, 50.35

while ur log : Overall AP40@easy, moderate, hard: bbox AP40:76.0208, 66.9024, 63.6359 bev AP40:73.5277, 63.4987, 59.6674 3d AP40:67.2419, 54.9371, 50.9578 aos AP40:65.59, 56.35, 53.32

Here is my config: sys.platform: linux Python: 3.8.8 (default, Apr 13 2021, 19:58:26) [GCC 7.3.0] CUDA available: True GPU 0,1,2,3,4,5,6,7: NVIDIA GeForce RTX 3090 CUDA_HOME: /usr/local/cuda-11.3 NVCC: Build cuda_11.3.r11.3/compiler.29745058_0 GCC: gcc (Ubuntu 9.4.0-1ubuntu1~20.04.2) 9.4.0 PyTorch: 1.10.0 PyTorch compiling details: PyTorch built with:

TorchVision: 0.11.0 OpenCV: 4.9.0 MMCV: 1.4.8 MMCV Compiler: GCC 9.4 MMCV CUDA Compiler: 11.3 MMDetection: 2.22.0 MMSegmentation: 0.22.1 MMDetection3D: 1.0.0rc0+a5dc465