RUC icon indicating copy to clipboard operation
RUC copied to clipboard

Can I train the model from scratch?

Open michaelzfm opened this issue 4 years ago • 2 comments

Hi, I found that if I don't use the preetrained model, the code don't work. So I can't train from scratch? If I need to train with other dataset like Imagenet, or use other backbone like resnet50, how can I find pretrained model? Thank you.

michaelzfm avatar Jul 07 '21 07:07 michaelzfm

yeah like for example we want to check for dogs and cats classification.so if we gave them as input,it will show type mismatch error while loading pretrained models weights and bias(cifar10) because they were trained on 10 classes.so what should we do now?

gauravjain10 avatar Jul 07 '21 10:07 gauravjain10

Hi, thank you for your interest in our work!

Our method is add-on module on top of the off-the-shelf clustering model. So I also borrowed pretrained model from SCAN github. https://github.com/wvangansbeke/Unsupervised-Classification If you want to train from scratch, please refer above SCAN github :)

Thank you again for your interest.

deu30303 avatar Jul 08 '21 01:07 deu30303