search

lxtGH / CAE

This is a PyTorch implementation of “Context AutoEncoder for Self-Supervised Representation Learning"
193 stars 22 forks source link

linear probe #5

Closed lywang76 closed 2 years ago

lywang76 commented 2 years ago

Do you have codes for linear probe?

SelfSup-MIM commented 2 years ago

Hi, we have uploaded code for linear probing before: https://github.com/lxtGH/CAE/blob/0916fa42a115fcb99007d9be8827a13828f310fb/scripts/cae_base_800e.sh#L40