Revert limit train batches

SeanNaren · SeanNaren · commit d20aea472348 · 2021-04-13T15:16:57.000+01:00
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,
     )
 

Original file line number	Diff line number	Diff line change
`@@ -122,7 +122,7 @@ def test_model_16bit_tpu_cores_1(tmpdir):`
`122`	`122`	`progress_bar_refresh_rate=0,`
`123`	`123`	`max_epochs=2,`
`124`	`124`	`tpu_cores=1,`
`125`		`- limit_train_batches=8,`
	`125`	`+ limit_train_batches=0.7,`
`126`	`126`	`limit_val_batches=2,`
`127`	`127`	`)`
`128`	`128`