added instructions to test

This commit is contained in:
William Falcon 2019-07-24 21:31:43 -04:00
parent 104b4dc1ff
commit a186cf12dc
2 changed files with 4 additions and 1 deletions

View File

@ -1,6 +1,9 @@
# Pytorch-Lightning Tests
## Running tests
The automatic travis tests ONLY run CPU-based tests. Although these cover most of the use cases,
run on a 2-GPU machine to validate the full test-suite.
To run all tests do the following:
```bash

View File

@ -163,7 +163,7 @@ def test_cpu_model_with_amp():
model, hparams = get_model()
with pytest.raises(ModuleNotFoundError):
with pytest.raises((MisconfigurationException, ModuleNotFoundError)):
run_gpu_model_test(trainer_options, model, hparams, on_gpu=False)