Debugging

lib/modules/utils.py

@@ -123,6 +123,10 @@ class LightningBaseModule(pl.LightningModule, ABC):
                           batch_size=self.hparams.data_param.batchsize,
                           num_workers=self.hparams.data_param.worker)
 
+    @property
+    def data_len(self):
+        return len(self.dataset.train_dataset)
+
     def configure_optimizers(self):
         raise NotImplementedError