From 9e5dd7a7eae3ae123b4a6c925c31631ebc4e89f9 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 17:02:39 -0400 Subject: [PATCH] added test for no dist sampler --- tests/test_models.py | 9 --------- 1 file changed, 9 deletions(-) diff --git a/tests/test_models.py b/tests/test_models.py index 22e7711d..23e3e3cc 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -231,19 +231,10 @@ def test_amp_gpu_ddp(): run_gpu_model_test(trainer_options, model, hparams) - - - # ------------------------------------------------------------------------ # UTILS # ------------------------------------------------------------------------ - def run_gpu_model_test(trainer_options, model, hparams, on_gpu=True): - """ - Make sure DDP + AMP work - :return: - """ - save_dir = init_save_dir() # exp file to get meta