akhaliq HF staff commited on
Commit
16bbf6a
1 Parent(s): 9e6247c

Update modules/lseg_module.py

Browse files
Files changed (1) hide show
  1. 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
- 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,
 
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,