mae-pytorch icon indicating copy to clipboard operation
mae-pytorch copied to clipboard
verified publisher icon catalys1

Downstream performance

Open matthias-brucklacher opened this issue 2 years ago • 2 comments

Nice stuff! Are you planning to evaluate the pretrained model on classification or detection tasks?

matthias-brucklacher avatar Jan 10 '24 16:01 matthias-brucklacher

I'd like to, I just haven't had the time to get to it. I'm also welcome to other people trying it out and sharing results!

catalys1 avatar Jan 25 '24 05:01 catalys1

I implemented a simple setting on the STL-10 dataset. I don't have experience with open source development, but would like to try creating a pull request a) to get practice and b) because people may be interested. It consists of new DataModules with config file, an extended model with a linear probe and of course a script to train and evaluate the model.

matthias-brucklacher avatar Jan 27 '24 11:01 matthias-brucklacher