From ef843d5f967e8220438969b1016e082c35fa3d31 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 13:56:21 -0400 Subject: [PATCH] refactored model tests --- tests/debug.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/debug.py b/tests/debug.py index 68aee841..095a060e 100644 --- a/tests/debug.py +++ b/tests/debug.py @@ -111,7 +111,7 @@ def main(): max_nb_epochs=1, gpus=[0, 1], distributed_backend='dp', - use_amp=True + use_amp=False ) result = trainer.fit(model)