Rayhane-mamah/Efficient-VDVAE

embeddings for TSNE

Odedbenc opened this issue · 0 comments

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.