Get latent feature space

Sara-Ahmed / SiT

Self-supervised vIsion Transformer (SiT)

324 stars 49 forks source link

Get latent feature space #25

Closed ramdhan1989 closed 2 years ago

ramdhan1989 commented 2 years ago

Hi, I would like to get latent feature space. From my understanding, I can take the output from here x = self.forward_features(x) in forward. am I correct ?

Thank you

Sara-Ahmed commented 2 years ago

Yes it is. Good luck!