search

ctr26 / bioimage_embed

3 stars 3 forks source link

Decoder fix #19

Closed ctr26 closed 8 months ago

ctr26 commented 8 months ago

Before the bolt-related decoders were using the size of the latent space rather than the native size of the output of resnet50's features. This meant that the decoder was trying to build a model different to the encoder. I think this fixes it. Fixed for vq and vae

ctr26 commented 8 months ago

@ctr26 @afoix

This is an almost fix. It fixes the param imbalance for resnet_18_[vae/vqvae] but not resnet_50_[vae/vqvae]

ctr26 commented 8 months ago

Don't merge yet as this is not passing testing will fix today