move backbones to root vision folder

Lightning-Universe · Feb 3, 2021 · 7f73823 · 7f73823
1 parent 41dd52e
commit 7f73823
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 57 deletions.
diff --git a/flash/vision/classification/backbones.py b/flash/vision/classification/backbones.py
diff --git a/flash/vision/embedding/image_embedder_model.py b/flash/vision/embedding/image_embedder_model.py
@@ -24,7 +24,7 @@
 from flash.core import Task
 from flash.core.data import TaskDataPipeline
 from flash.core.data.utils import _contains_any_tensor
-from flash.vision.classification.backbones import torchvision_backbone_and_num_features
+from flash.vision.backbones import torchvision_backbone_and_num_features
 from flash.vision.classification.data import _default_valid_transforms, _pil_loader
 from flash.vision.embedding.model_map import _load_bolts_model, _models