diff --git a/train.py b/train.py
index bc40843..2812a6b 100644
--- a/train.py
+++ b/train.py
@@ -45,8 +45,8 @@ def main():
         # benchmark=False,
         # default_root_dir="lightning_logs",
         # detect_anomaly=True,
-        limit_train_batches=80,
-        limit_val_batches=40,
+        # limit_train_batches=80,
+        # limit_val_batches=40,
         # reload_dataloaders_every_n_epochs=10,
     )