From ffdf11b7edb1af86bb453b7566bab43e0c7306b9 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 17:35:39 -0400 Subject: [PATCH] test memory printing --- tests/test_models.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_models.py b/tests/test_models.py index c958b0d0..4195f9a6 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -36,9 +36,6 @@ def test_cpu_model(): model, hparams = get_model() - # test memory gathering - model.count_mem_items() - run_gpu_model_test(trainer_options, model, hparams, on_gpu=False) @@ -237,6 +234,9 @@ def test_ddp_sampler_error(): use_amp=True ) + # test memory gathering + trainer.count_mem_items() + with pytest.raises(MisconfigurationException): trainer.get_dataloaders(model)