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

Will the training code be open-sourced? When will it be released? #5

Open
CIawevy opened this issue Oct 14, 2024 · 3 comments
Open

Comments

@CIawevy
Copy link

CIawevy commented Oct 14, 2024

Hi,
I just wanted to say that I really appreciate the great work on this project! I’m curious if the training code will be open-sourced. If so, do you have an estimated timeline for its release?
Thanks!

@CIawevy
Copy link
Author

CIawevy commented Oct 14, 2024

Or could you recommend me a code base for reproducing flowGen,Did you use the pix2pix code base or not?

@alex4727
Copy link
Collaborator

Hi, sorry for the late reply; I've been really busy lately :(

Regarding your previous question, fine-tuning has to be done for each specific scene (with a relevant video). As mentioned in the appendix of the paper, we later observed that initializing with the base CelebV-Text trained model resulted in similar results as SINTEL & FlyingChair initialized ones. And fine-tuning approach requires 1) careful data selection & preprocessing and 2) additional training. So based on our empirical results, the generalizability of the base models is usually more useful than the fine-tuning approach (which heavily depends on the data selection and requires additional training).

While we currently do plan to release the training code for reference, please understand that it could take a while (like December or later) as we are working on other projects and are really busy these days :(

About the second question, yes, you can use the Pix2Pix codebase. We have optimized it to get it running with the HF accelerator environment with larger batches and multi-GPU support, but the Pix2Pix codebase is a good starting point.

Thanks for your interest in our work!

@CIawevy
Copy link
Author

CIawevy commented Oct 14, 2024

Thanks for your response!

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