search

princeton-vl / RAFT-3D

BSD 3-Clause "New" or "Revised" License
233 stars 26 forks source link

Repro for RAFT variants #4

Open SamriddhiJain opened 3 years ago

SamriddhiJain commented 3 years ago

Hi

Thanks for sharing the work. I'm trying to repro the RAFT variants (2D flow backprojected, 3D flow) mentioned in the paper. I'm facing some numerical instabilities and was wondering did you experience anything like that? And if yes, did you try some different learning schedules as of RAFT? Would it be possible to share that code?

Thanks

zachteed commented 3 years ago

Hi, I used the same learning rate for all the ablations of the RAFT variants. I don't remember having any issues with numerical instability with these models. Are you having issues with the training of the model included in the repo, or just the variants mentioned in the paper?

I can provide code for the 3D and 2D-backproj in 1-2 days, just need to check that they still work fine with the refactored code

SamriddhiJain commented 3 years ago

Are you having issues with the training of the model included in the repo, or just the variants mentioned in the paper?

Just the variants right now.

I can provide code for the 3D and 2D-backproj in 1-2 days, just need to check that they still work fine with the refactored code

Great, thanks.

SamriddhiJain commented 3 years ago

Hey, any updates on this?