salesforce / ULIP

BSD 3-Clause "New" or "Revised" License
431 stars 42 forks source link

Can not load pretrained weight of PointNext in finetuning on ModelNet40, in_channels is 3 not 4 #39

Open TangYuan96 opened 1 year ago

TangYuan96 commented 1 year ago

First of all, thank you for sharing such an excellent work !!

The pretrained weight of PointNext is released, where '-- use_height' is used. However, during finetune on ModelNet40 or ScanObjectNN, there was no "-- use_height" in the data, which resulted in the pretrained weight being unable to be loaded.

The in_channels is 4 in pretrained weight of PointNext. But the in_channels is 3 in setting of finetune on ModelNet40 or ScanObjectNN

Tycho-Xue commented 1 year ago

@TangYuan96 , sorry for the late reply, I was out of the office until this week. I think by default we use --use_height during the pre-training since it typically gets the best performance, I believe during the finetuing, you can also enable the 4 channel input, that's how we fine-tuned the model.