Remove accidental training batch limit

Dawars · Sep 26, 2023 · b6175bb · b6175bb
1 parent c171331
commit b6175bb
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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"]