Merge branch 'master' into update/codeowners

Lightning-Universe · Aug 16, 2021 · 9e6da35 · 9e6da35
2 parents d745aa1 + 0e8c0ce
commit 9e6da35
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flash/image/embedding/model.py b/flash/image/embedding/model.py
@@ -107,7 +107,7 @@ def forward(self, x) -> torch.Tensor:
         if isinstance(x, tuple):
             x = x[-1]
 
-        if x.dim() == 4 and self.embedding_dim:
+        if x.dim() == 4 and not self.embedding_dim:
             x = self.apply_pool(x)
 
         x = self.head(x)