fixed supervisely (#1603)

Co-authored-by: Ofri Masad <ofrimasad@users.noreply.github.com>
Deci-AI · Nov 5, 2023 · 4d3abe1 · 4d3abe1
1 parent b90f4d8
commit 4d3abe1
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/src/super_gradients/training/dataloaders/dataloaders.py b/src/super_gradients/training/dataloaders/dataloaders.py
@@ -31,7 +31,6 @@
     CoCoSegmentationDataSet,
     PascalVOC2012SegmentationDataSet,
     PascalVOCAndAUGUnifiedDataset,
-    SuperviselyPersonsDataset,
     MapillaryDataset,
 )
 from super_gradients.training.utils import get_param
@@ -755,7 +754,6 @@ def pascal_voc_segmentation_val(dataset_params: Dict = None, dataloader_params:
 def supervisely_persons_train(dataset_params: Dict = None, dataloader_params: Dict = None) -> DataLoader:
     return get_data_loader(
         config_name="supervisely_persons_dataset_params",
-        dataset_cls=SuperviselyPersonsDataset,
         train=True,
         dataset_params=dataset_params,
         dataloader_params=dataloader_params,
@@ -766,7 +764,6 @@ def supervisely_persons_train(dataset_params: Dict = None, dataloader_params: Di
 def supervisely_persons_val(dataset_params: Dict = None, dataloader_params: Dict = None) -> DataLoader:
     return get_data_loader(
         config_name="supervisely_persons_dataset_params",
-        dataset_cls=SuperviselyPersonsDataset,
         train=False,
         dataset_params=dataset_params,
         dataloader_params=dataloader_params,