facebookresearch / VLPart

[ICCV2023] VLPart: Going Denser with Open-Vocabulary Part Segmentation
MIT License
348 stars 16 forks source link

Unstable training when saving zero-shot weights #9

Closed joshmyersdean closed 9 months ago

joshmyersdean commented 9 months ago

Hello!

I am trying to reproduce the pascal part results when going from base to novel categories and am experiencing different results from saving my own weights vs. using the ones you have provided in datasets/metadata.

For example, when using the saved weights you provide these are my results after 2000 iterations (1 gpu).

image

And these are my results after saving new weights using tools/pascal_part_base_clip_name.py. image

The only difference seems to be in the precision between these weights (in the 1e-4 place). Do you know why there would be such a discrepancy?

Thank you! Josh