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.

Sign up for GitHub

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

Jump to bottom

Code changes to finetune PointBERT or ACT #5

Open

yhxu022 opened this issue Oct 19, 2023 · 1 comment

yhxu022 commented Oct 19, 2023

What changes should be made to the code if I want to finetune PointBERT or ACT pre-trained models?

Owner

zyh16143998882 commented Oct 19, 2023

IDPT is easily migrated to Point-BERT or other models. Only the following modifications are required:

add "part: only_new," to the .yaml file in the cfg directory, for example:

optimizer : {
  type: AdamW,
  part: only_new,
  kwargs: {
  lr : 0.0005,
  weight_decay : 0.05
}}

Modify the "add_weight_decay" function in the "build_opti_sche function" with reference to here
Modify the Transformer backbone to include code of the Dynamic Prompt Generation Module(here)

zyh16143998882 mentioned this issue

Open

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment