Training Code

Hi **@wenbowen123**, thank you for your great work and for open-sourcing this project! 

I would like to fine-tune the model on my own stereo dataset. However, I noticed the repository only includes inference code and does not contain training scripts.  Could you consider releasing the training code? If you have plans to release it, is there an estimated timeline?

Thank you for your time!