search

yifanlu0227 / HEAL

[ICLR2024] HEAL: An Extensible Framework for Open Heterogeneous Collaborative Perception ➡️ All You Need for Multi-Modality Collaborative Perception!
Other
147 stars 9 forks source link

Question about DiscoNet #5

Open Xu2729 opened 6 months ago

Xu2729 commented 6 months ago

Hello! Thank you for your excellent work!

I encountered an issue while trying to run the DiscoNet model on the DAIR-V2X dataset using the command python opencood/tools/train.py --hypes_yaml opencood/hypes_yaml/dairv2x/LiDAROnly/lidar_disco.yaml. The error message ModuleNotFoundError: No module named 'opencood.models.fuse_modules.disco_fuse' suggests that the opencood\models\fuse_modules\disco_fuse.py file is missing. Could you please provide this file?

Additionally, I noticed that there is a train_w_kd.py script for distillation. Could you also provide the configuration for the teacher model (early fusion) and the corresponding command for training DiscoNet using this teacher model?

I would greatly appreciate your prompt assistance.

yifanlu0227 commented 6 months ago

Currently i am busy with other projects but you can find the configuration file in my CoAlign repo. Could you please first check it ?

Xu2729 commented 6 months ago

Thank you for your prompt reply, I will give it a try.