From b6175bb25e02e6c422e70d0bfcadde8cfa2b7ef7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?D=C3=A1vid=20Komorowicz?= <david.komorowicz@gini.net>
Date: Tue, 26 Sep 2023 12:56:31 +0200
Subject: [PATCH] Remove accidental training batch limit

---
 docmae/train.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/docmae/train.py b/docmae/train.py
index a0b805d..319d395 100644
--- a/docmae/train.py
+++ b/docmae/train.py
@@ -75,7 +75,6 @@ def train(args, config: dict, datamodule: L.LightningDataModule):
         max_steps=config["training"].get("steps", -1),
         num_sanity_val_steps=1,
         enable_progress_bar=config["progress_bar"],
-        limit_train_batches=20
     )
 
     hidden_dim = config["model"]["hidden_dim"]