akhaliq HF staff commited on
Commit
ea40b51
1 Parent(s): 9c58f57

Update modules/lseg_module.py

Browse files
Files changed (1) hide show
  1. 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
- self.num_classes = 255
62
- self.train_accuracy = pl.metrics.Accuracy()
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,