[fix] resnet50_vae duplicated

uhlmanngroup · Oct 1, 2024 · 5dddb33 · 5dddb33
1 parent 335ea77
commit 5dddb33
Showing 1 changed file with 3 additions and 16 deletions.
diff --git a/bioimage_embed/models/factory.py b/bioimage_embed/models/factory.py
@@ -18,6 +18,7 @@
 from . import bolts
 from functools import partial
 
+
 class ModelFactory:
     def __init__(
         self, input_dim, latent_dim, pretrained=False, progress=True, **kwargs
@@ -136,26 +137,13 @@ def resnet18_beta_vae(self):
                 pythae.models.BetaVAEConfig,
                 use_default_encoder=False,
                 use_default_decoder=False,
-                **self.kwargs
+                **self.kwargs,
             ),
             pythae.models.BetaVAE,
             bolts.ResNet18VAEEncoder,
             bolts.ResNet18VAEDecoder,
         )
 
-    def resnet50_vae(self):
-        return self.create_model(
-            partial(
-                pythae.models.VAEConfig,
-                use_default_encoder=False,
-                use_default_decoder=False,
-                **self.kwargs
-            ),
-            pythae.models.VAE,
-            bolts.ResNet50VAEEncoder,
-            bolts.ResNet50VAEDecoder,
-        )
-
     def resnet50_vqvae(self):
         return self.create_model(
             partial(
@@ -175,7 +163,7 @@ def resnet50_beta_vae(self):
                 pythae.models.BetaVAEConfig,
                 use_default_encoder=False,
                 use_default_decoder=False,
-                **self.kwargs
+                **self.kwargs,
             ),
             pythae.models.BetaVAE,
             bolts.ResNet50VAEEncoder,
@@ -257,7 +245,6 @@ def __call__(self, model):
 ]
 
 
-
 def create_model(
     model: str,
     input_dim: Tuple[int, int, int],