Update modules/lseg_module.py

modules/lseg_module.py

@@ -52,12 +52,12 @@ class LSegModule(LSegmentationModule):
         self.train_transform = transforms.Compose(train_transform)
         self.val_transform = transforms.Compose(val_transform)
 
-        self.trainset = self.get_trainset(
-            dataset,
-            augment=kwargs["augment"],
-            base_size=self.base_size,
-            crop_size=self.crop_size,
-        )
+       # self.trainset = self.get_trainset(
+        #    dataset,
+       #     augment=kwargs["augment"],
+       #     base_size=self.base_size,
+       #     crop_size=self.crop_size,
+       # )
         
         self.valset = self.get_valset(
             dataset,