Update modules/lseg_module.py

modules/lseg_module.py

@@ -58,6 +58,8 @@ class LSegModule(LSegmentationModule):
        #     base_size=self.base_size,
        #     crop_size=self.crop_size,
        # )
+       self.num_classes = 255
+       self.train_accuracy = pl.metrics.Accuracy()
         
         self.valset = self.get_valset(
             dataset,