diff --git a/tests/models/test_tpu.py b/tests/models/test_tpu.py index 4b7c127bf515fe..92c336e0acf39d 100644 --- a/tests/models/test_tpu.py +++ b/tests/models/test_tpu.py @@ -173,6 +173,7 @@ def test_model_16bit_tpu_index_5(tmpdir): model = EvalModelTemplate() trainer = Trainer( default_root_dir=tmpdir, + precision=16, max_epochs=1, train_percent_check=0.4, val_percent_check=0.2,