search

cvlab-columbia / CT4Recognition

22 stars 8 forks source link

How to run Supervised ResNet experiments from scratch? #1

Open JeanKaddour opened 2 years ago

JeanKaddour commented 2 years ago

Hi Chengzhi and Kevin,

Thank you so much for this great repo!

I'm wondering how to best reproduce the supervised ResNet experiments?

I want to train the ResNet from scratch, not using an SSL-pre-trained one.

Basically, I'd like to reproduce the entries ["ResNet50", "ResNet152", "ResNet101-2x"] of Table 4, but it's not clear to me how to do so exactly given your code, i.e., what script I have to run with what arguments.

Thanks a lot in advance.

ChengzhiCU commented 2 years ago

Thank you for your interest in our work!

It is not pushed yet. We download the official model from pytorch [torchvision] for ["ResNet50", "ResNet152", "ResNet101-2x"] , and replace the SSL model, then everything else is the same as SSL. Right now you need some modification of the code. Will push that in a few days as I am traveling.