Update modules/lseg_module.py
Browse files- modules/lseg_module.py +6 -6
modules/lseg_module.py
CHANGED
@@ -52,12 +52,12 @@ class LSegModule(LSegmentationModule):
|
|
52 |
self.train_transform = transforms.Compose(train_transform)
|
53 |
self.val_transform = transforms.Compose(val_transform)
|
54 |
|
55 |
-
|
56 |
-
|
57 |
-
|
58 |
-
|
59 |
-
|
60 |
-
|
61 |
|
62 |
self.valset = self.get_valset(
|
63 |
dataset,
|
|
|
52 |
self.train_transform = transforms.Compose(train_transform)
|
53 |
self.val_transform = transforms.Compose(val_transform)
|
54 |
|
55 |
+
# self.trainset = self.get_trainset(
|
56 |
+
# dataset,
|
57 |
+
# augment=kwargs["augment"],
|
58 |
+
# base_size=self.base_size,
|
59 |
+
# crop_size=self.crop_size,
|
60 |
+
# )
|
61 |
|
62 |
self.valset = self.get_valset(
|
63 |
dataset,
|