-
Notifications
You must be signed in to change notification settings - Fork 37
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
The PCK on pf-pascal dataset is 75.35 #7
Comments
Could you share your env setting? PyTorch version? system version?
I got similar results with yours in stage 1. But I can only get 76.50% after the finetune stage. Because of the lack of the hyperparameter of stage 2, I have finetuned all blocks of the last residual layer as the paper present. @ignacio-rocco Would you like to share the more training details with us. Thanks a lot! |
According to checkpoint['args'], the fe_finetune_params is set to 1. |
I use the multi-gpu(2-gpus), set batch size =32, get 77.3% for the first stage.
I will report the performance and the log later. |
Run
python train.py --ncons_kernel_sizes 5 5 5 --ncons_channels 16 16 1 --dataset_image_path datasets/pf-pascal --dataset_csv_path datasets/pf-pascal/image_pairs/
The PCK on pf-pascal dataset is 75.35 (78.9 in paper).
Is there some other important hyperparameters? Thank you.
The text was updated successfully, but these errors were encountered: