Register
Login
Resources
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Product
Data Engine LLMs Platform Enterprise
Pricing Explore
Connect to our Discord channel

#716 Fix mapilliary_dataset yaml

Merged
Ghost merged 1 commits into Deci-AI:master from deci-ai:hotfix/SG-000-fix_mapilliary_dataset
@@ -4,7 +4,7 @@ config_ignore_values:
   '1.2': 65
   '1.2': 65
   '2.0': 123
   '2.0': 123
 
 
-mask_fill_value: ${getitem:${dataset_params.config_ignore_values,${dataset_params.config_version}}}
+mask_fill_value: ${getitem:${dataset_params.config_ignore_values},${dataset_params.config_version}}
 
 
 train_dataset_params:
 train_dataset_params:
   root_dir: /data/mapillary-vistas-dataset_public_v2.0
   root_dir: /data/mapillary-vistas-dataset_public_v2.0
Discard
@@ -60,6 +60,7 @@ import numpy as np
 
 
 from super_gradients.training.datasets.detection_datasets.pascal_voc_detection import PascalVOCUnifiedDetectionTrainDataset
 from super_gradients.training.datasets.detection_datasets.pascal_voc_detection import PascalVOCUnifiedDetectionTrainDataset
 from super_gradients.training.datasets.segmentation_datasets import MapillaryDataset
 from super_gradients.training.datasets.segmentation_datasets import MapillaryDataset
+from super_gradients import init_trainer
 
 
 
 
 @register_dataset("FixedLenDataset")
 @register_dataset("FixedLenDataset")
@@ -71,6 +72,9 @@ class FixedLenDataset(TensorDataset):
 
 
 
 
 class DataLoaderFactoryTest(unittest.TestCase):
 class DataLoaderFactoryTest(unittest.TestCase):
+    def setUp(self) -> None:
+        init_trainer()
+
     def test_coco2017_train_creation(self):
     def test_coco2017_train_creation(self):
         dl_train = coco2017_train()
         dl_train = coco2017_train()
         self.assertTrue(isinstance(dl_train, DataLoader))
         self.assertTrue(isinstance(dl_train, DataLoader))
Discard