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

my model performed poor #159

Open
sdpscnc opened this issue Oct 16, 2024 · 1 comment
Open

my model performed poor #159

sdpscnc opened this issue Oct 16, 2024 · 1 comment

Comments

@sdpscnc
Copy link

sdpscnc commented Oct 16, 2024

I tested my self trained model with epoch=25 step=398632.ckpt on the LFW dataset and found poor performance, with a maximum similarity score of only 0.1-0.2. However, the pre trained model adaface_ir50_ms1mv2.ckpt performed well with a maximum similarity score of 0.99. May I ask where the problem may be?
I used run_ir50_ms1mv2.sh for training:
python main.py
--data_root /home/xxx/PycharmProjects/AdaFace/data
--train_data_path faces_webface
--val_data_path faces_webface
--prefix ir50_ms1mv2_adaface
--gpus 1
--use_16bit
--arch ir_50
--batch_size 32
--num_workers 16
--epochs 26
--lr_milestones 12,20,24
--lr 0.1
--head adaface
--m 0.4
--h 0.333
--low_res_augmentation_prob 0.2
--crop_augmentation_prob 0.2
--photometric_augmentation_prob 0.2

@afm215
Copy link

afm215 commented Jan 31, 2025

Hello, on which dataset are you training on exactly? Be aware that you should set the parameter --custom_num_class as well

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