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 model by my own custom dataset. #175

Open
sangtv9 opened this issue Aug 12, 2021 · 3 comments
Open

Fine tuning model by my own custom dataset. #175

sangtv9 opened this issue Aug 12, 2021 · 3 comments

Comments

@sangtv9
Copy link

sangtv9 commented Aug 12, 2021

Thanks for you great repo. I used pretrained model for my dataset. But, It does not reach good enough accuracy. So I must re-train the model by my dataset. But I only reach 65% at accuracy on validation dataset. The model loss only decrease to approximate 10. Have anyone met the issue? Have you had any idea to deal this? Thanks very much.

@sangtv9
Copy link
Author

sangtv9 commented Aug 12, 2021

This is my config for training
SGD ( Parameter Group 0 dampening: 0 lr: 0.05 momentum: 0.9 nesterov: False weight_decay: 0.0004 Parameter Group 1 dampening: 0 lr: 0.05 momentum: 0.9 nesterov: False weight_decay: 0 )
I use CrossEntropy for loss function

@momohuangsha
Copy link

@sangtv9 Hey! Has your problem been solved? whta is your backbone network? I have the same problem (backbone:mobileface)

@ANDRESHZ
Copy link

Hi @momohuangsha and @sangtv9, I hope you are having nice results, I am wondering if you finally found a way to tune with your dataset.

Can you share with the community your code?

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