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

Hyperparameters of several finetuning experiments #83

Open
wwsource opened this issue Mar 23, 2023 · 2 comments
Open

Hyperparameters of several finetuning experiments #83

wwsource opened this issue Mar 23, 2023 · 2 comments

Comments

@wwsource
Copy link

wwsource commented Mar 23, 2023

Thank you for your great work!

Could you give the hyperparameters (like learning rate, batch size, crop size, and step number) that can reproduce the paper data in three finetuning experiments (KITTI, ETH3D and Middlebury), which are not explicitly described in the paper "RAFT-Stereo: Multilevel Recurrent Field Transforms for Stereo Matching".

Thank you very much.

@lahavlipson
Copy link
Collaborator

These are the settings:

Middlebury: lr=2e-5, batch_size 2, crop_size=384x1000, steps=4k
ETH3D: lr=1e-5, batch_size 2, crop_size=384x1000, steps=4k saturation_range=[0 0] img_gamma=[0.5 2.0 0.8 1.2]
KITTI: lr=1e-5, batch_size 3, crop_size=320x1000, steps=5k

@rmadan2
Copy link

rmadan2 commented Sep 24, 2023

Thank you for your response above.
Can you also please specify the number of disparity field updates configured for fine-tuning and inference, for all three datasets? Thank you.

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

3 participants