3dlg-hcvc / M3DRef-CLIP

[ICCV 2023] Multi3DRefer: Grounding Text Description to Multiple 3D Objects
https://3dlg-hcvc.github.io/multi3drefer/
MIT License
64 stars 3 forks source link

RuntimeError when setting "model.network.detector.use_gt_proposal=True" #7

Closed Xiaolong-RRL closed 9 months ago

Xiaolong-RRL commented 10 months ago

Dear author:

Thanks for your interesting work.

When I run the following command:

# train a model with the pretrained detector, using GT object proposals
python train.py data={scanrefer/nr3d/multi3drefer} experiment_name={any_string} +detector_path=checkpoints/PointGroup_ScanNet.ckpt model.network.detector.use_gt_proposal=True

an error has occurred:

RuntimeError: It looks like your LightningModule has parameters that were not used in producing the loss returned by training_step. If this is intentional, you must enable the detection of unused parameters in DDP, either by setting the string value `strategy='ddp_find_unused_parameters_true'` or by setting the flag in the strategy with `strategy=DDPStrategy(find_unused_parameters=True)`.

It seems like the model return some variables that do not calculate loss, and I wander how to solve it?

Best! Xiaolong

eamonn-zh commented 9 months ago

This should have nothing to do with the use_gt_proposal arg, could you please share which lightning version you are using?

Xiaolong-RRL commented 9 months ago

The lightning version I use is 2.0.6

eamonn-zh commented 9 months ago

We haven't tested training the model on multiple GPUs.