From b47b881f788ff210565c977028e3ab0d8f66dab1 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 23 Oct 2019 06:13:00 -0400 Subject: [PATCH] Update README.md --- tests/README.md | 18 ++---------------- 1 file changed, 2 insertions(+), 16 deletions(-) diff --git a/tests/README.md b/tests/README.md index 032cf04f..9cf8af78 100644 --- a/tests/README.md +++ b/tests/README.md @@ -1,4 +1,6 @@ # PyTorch-Lightning Tests +Most PL tests train a full MNIST model under various trainer conditions (ddp, ddp2+amp, etc...). +This provides testing for most combinations of important settings. ## Running tests The automatic travis tests ONLY run CPU-based tests. Although these cover most of the use cases, @@ -26,22 +28,6 @@ The GPU machine must have: 2. [NVIDIA-apex](https://github.com/NVIDIA/apex#linux) installed. -### test_models.py -This file fits a tiny model on MNIST using these different set-ups. -1. CPU only. -2. Single GPU with DP. -3. Multiple (2) GPUs using DP. -3. Multiple (2) GPUs using DDP. -3. Multiple (2) GPUs using DP + apex (for 16-bit precision). -3. Multiple (2) GPUs using DDP + apex (for 16-bit precision). - -For each set up it also tests: -1. model saving. -2. model loading. -3. predicting with a loaded model. -4. simulated save from HPC signal. -5. simulated load from HPC signal. - ## Running Coverage Make sure to run coverage on a GPU machine with at least 2 GPUs and NVIDIA apex installed.