Populate default_preprocess in embedding model

Lightning-Universe · Apr 14, 2021 · 3ed7b82 · 3ed7b82
1 parent 5e0d0ca
commit 3ed7b82
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/flash/vision/embedding/model.py b/flash/vision/embedding/model.py
@@ -49,10 +49,6 @@ class ImageEmbedder(Task):
 
     backbones: FlashRegistry = IMAGE_CLASSIFIER_BACKBONES
 
-    @property
-    def preprocess(self):
-        return ImageClassificationPreprocess(predict_transform=ImageClassificationData.default_val_transforms())
-
     def __init__(
         self,
         embedding_dim: Optional[int] = None,
@@ -70,6 +66,9 @@ def __init__(
             optimizer=optimizer,
             metrics=metrics,
             learning_rate=learning_rate,
+            default_preprocess=ImageClassificationPreprocess(
+                predict_transform=ImageClassificationData.default_val_transforms(),
+            )
         )
 
         self.save_hyperparameters()