Revert limit train batches

Lightning-AI · Apr 13, 2021 · d20aea4 · d20aea4
1 parent 39c1632
commit d20aea4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/test_tpu.py b/tests/models/test_tpu.py
@@ -122,7 +122,7 @@ def test_model_16bit_tpu_cores_1(tmpdir):
         progress_bar_refresh_rate=0,
         max_epochs=2,
         tpu_cores=1,
-        limit_train_batches=8,
+        limit_train_batches=0.7,
         limit_val_batches=2,
     )