embeddings for TSNE - Githubissues

Rayhane-mamah / Efficient-VDVAE

Official Pytorch and JAX implementation of "Efficient-VDVAE: Less is more"

https://arxiv.org/abs/2203.13751

MIT License

190 stars 23 forks source link

embeddings for TSNE #17

Open Odedbenc opened 9 months ago

Odedbenc commented 9 months ago

Hi, I tried your model and it is super impressive. I want to use the embeddings for TSNE plots. When I run the encode mode, I get a dict with all the layers per image. Is there a way to obtain the last embedding vector from the encoder? Thanks.