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

Why is the fine-tuning result on modelnet40 obtained by my pre-trained model so different from the pre-trained model you gave? #14

Open
ProgrammerSakura opened this issue Apr 28, 2024 · 0 comments

Comments

@ProgrammerSakura
Copy link

Here are the results of my fine-tuning:

Model parameter loading process:
image
image

The best result I got from reproducing fine-tuning:
image

You gave:

The model parameter loading process I got using the pre-trained model you gave:
image

The best result obtained from fine-tuning:
image

The model parameter loading process is inconsistent. Is it a problem with my pre-training?

The training step script i used is :
Pretraining got the pretarin model ckpt-best.pth
python main.py --config cfgs/pre-training/point-m2ae.yaml --exp_name pre-train

Then i copy this ckpt-best.pth to ckpts document, and use this scipt to funtune on modelnet40:
python main.py --config cfgs/fine-tuning/modelnet40.yaml --finetune_model --exp_name finetune --ckpts ckpts/ckpt-best.pth

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

1 participant