search

ctr26 / bioimage_embed

3 stars 3 forks source link

Bottleneck feature vector size #2

Closed spectralnanodiamond closed 8 months ago

spectralnanodiamond commented 1 year ago

It would be great to have bottleneck feature vector size of the VAE to be an accessible parameter. This would be help it be changed so that it was reasonable compared to the input patch size.

As part of this it may be necessary to have number of layers increase or decrease to keep the VAE network contraction and then expansion smooth. This could be another parameter, or calculated automatically.

ctr26 commented 1 year ago

https://github.com/ctr26/bio_vae/pull/3

Adding pythae compatibility which gives you easier access to latent space size

ctr26 commented 8 months ago

Resolved for most models pythae and legacy