search

Cccccczh404 / H-SAM

MIT License
76 stars 5 forks source link

About num_classes #9

Closed zzzyzh closed 5 months ago

zzzyzh commented 5 months ago

Hi, I noticed that your code doesn't seem to be adapting well to the num_classes parameter! When I tested your method on my own dataset (a dataset with only 4 classes), it didn't work properly, could you please refine the code for this?

zzzyzh commented 5 months ago

I have fixed this bug and open a PR. Please check it!

Cccccczh404 commented 5 months ago

Thanks very much for your interest and contribution! I have merged this commit.