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

Fine tuning #19

Open
laugam opened this issue Sep 28, 2022 · 4 comments
Open

Fine tuning #19

laugam opened this issue Sep 28, 2022 · 4 comments

Comments

@laugam
Copy link

laugam commented Sep 28, 2022

Hi, thank you for making your code available. The latest model release (v4.6) works really well for my project.
I would like to fine tune the model on my dataset.
Do you plan to make the script for the fine tuning available any time soon?
Thank you.

@hzwer
Copy link
Owner

hzwer commented Oct 1, 2022

https://github.com/megvii-research/ECCV2022-RIFE
I think if you can train original RIFE, it will be not hard to finetune practical-RIFE v4.6. If you meet any problem, just raise the issue.

@laugam
Copy link
Author

laugam commented Oct 7, 2022

Thank you for your reply.
The train and evaluate functions that you defined in the file train.py don't seem to be compatible with the new files RIFE_HDv3.py and IFNet_HDv3.py, for instance the loss functions are different.
Do you have any suggestions on how to modify those functions?

Many thanks.

@xsp1
Copy link

xsp1 commented Jul 3, 2023

Hello, I am also trying to retrain or fine-tune the v4.6 model on my own data set recently. I also found that RIFE_HDv3.py and IFNet_HDv3.py are different from the original version. I modified a lot of codes, such as removing After using loss_cons, I found that the effect is very poor after successful training. Did you fine-tune successfully? How did you solve these problems? Thanks!

@laugam
Copy link
Author

laugam commented Jul 6, 2023

Hello, unfortunately I did not manage to train/fine-tune the v4.6 model, I'm using the model as it is for the moment.
Maybe you will find useful the comments on this issue: hzwer/ECCV2022-RIFE#293 (comment)

I asked for the training code there but they didn't reply yet.
Please let me know if you succeed in this.
Thanks!

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