Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Open
TangYuan96 opened this issue Jul 26, 2023 · 1 comment

Comments

@TangYuan96
Copy link

TangYuan96 commented Jul 26, 2023

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
Copy link
Collaborator

@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.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants