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

Could you please give me training instructions #6

Closed
linxcan opened this issue Nov 15, 2022 · 2 comments
Closed

Could you please give me training instructions #6

linxcan opened this issue Nov 15, 2022 · 2 comments

Comments

@linxcan
Copy link

linxcan commented Nov 15, 2022

Hello!Thank you for your work. I am very interested in your work. Could you please give me training instructions? I want to try to train. Thank you very much.

@mbanani
Copy link
Owner

mbanani commented Nov 17, 2022

Hi @linxcan,

Thank you for your interest and sorry I never updated the train docs. The config defaults should be the same one as the main experiment. So running the following command should work once you set up the environment and dataset appropriately. To train a base model on ScanNet, you can just use the debug.yaml config as follows:

python train.py train debug.yaml

You could define experiment groups in folder with modification configs, check this portion of the code:

unsupervisedRR/train.py

Lines 36 to 40 in 4ba5b1c

# deal with modifiers
if args.modifier is not None:
assert cfg.EXPERIMENT.name == "", "Modifier config defines experiment name."
modifier_path = os.path.join("unsupervisedRR/configs", args.modifier)
cfg.merge_from_file(modifier_path)

If you are interested in unsupervised registration, I would suggest checking my follow-up work byoc which builds on this work and simplifies it by removing the rendering component and achieving better performance. If you're interested in learning visual features, you should use the vis_byoc config. You should also check issue #5 for some tips and tricks to improve performance.

Please let me know if you have any other questions.

@linxcan
Copy link
Author

linxcan commented Nov 17, 2022

Thank you for your reply. It helps me a lot.

@mbanani mbanani closed this as completed Nov 17, 2022
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