Update modules/lseg_module.py
Browse files- modules/lseg_module.py +2 -2
modules/lseg_module.py
CHANGED
@@ -58,8 +58,8 @@ class LSegModule(LSegmentationModule):
|
|
58 |
# base_size=self.base_size,
|
59 |
# crop_size=self.crop_size,
|
60 |
# )
|
61 |
-
|
62 |
-
|
63 |
|
64 |
#self.valset = self.get_valset(
|
65 |
# dataset,
|
|
|
58 |
# base_size=self.base_size,
|
59 |
# crop_size=self.crop_size,
|
60 |
# )
|
61 |
+
self.num_classes = 255
|
62 |
+
self.train_accuracy = pl.metrics.Accuracy()
|
63 |
|
64 |
#self.valset = self.get_valset(
|
65 |
# dataset,
|