About .pth file that computer_center.py need

zhangdoudou123 / SemFew

[CVPR2024] Simple Semantic-Aided Few-Shot Learning

18 stars 4 forks source link

About .pth file that computer_center.py need #4

Open KZF-kzf opened 1 month ago

KZF-kzf commented 1 month ago

Hello, I am executing the computecenter.py file, and I noticed that he needs a backbone{Dataset}.pth file, is it the weight of SwinT/ResNet-{Dataset}.pth that you gave? Are these weights obtained by pre-training on the base set? Looking forward to your answer.

zhangdoudou123 commented 1 month ago

Yes, you are right. These weights are obtained by base split training and are used to compute approximate class centers.

KZF-kzf commented 1 month ago

Thank you very much for your answer. I tested according to the weight, but can't repetition you on MiniImageNet results: https://github.com/zhangdoudou123/SemFew/issues/5. Is there a problem with my lab setup?