COCO dataset setting2

Iywie · Jan 12, 2023 · 9925d3e · 9925d3e
1 parent 0966b41
commit 9925d3e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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,
     )