Update modules/lseg_module.py

modules/lseg_module.py

@@ -61,12 +61,12 @@ class LSegModule(LSegmentationModule):
        self.num_classes = 255
        self.train_accuracy = pl.metrics.Accuracy()
         
-        self.valset = self.get_valset(
-            dataset,
-            augment=kwargs["augment"],
-            base_size=self.base_size,
-            crop_size=self.crop_size,
-        )
+        #self.valset = self.get_valset(
+        #    dataset,
+        #    augment=kwargs["augment"],
+        #    base_size=self.base_size,
+        #    crop_size=self.crop_size,
+        #)
 
         use_batchnorm = (
             (not kwargs["no_batchnorm"]) if "no_batchnorm" in kwargs else True