From 665c1507f035d7777a5ce9c414c0ba73e120fb74 Mon Sep 17 00:00:00 2001 From: Jirka Borovec Date: Thu, 13 Aug 2020 12:29:27 +0200 Subject: [PATCH] deterministic=True (#2944) --- tests/core/test_datamodules.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/tests/core/test_datamodules.py b/tests/core/test_datamodules.py index 0dc81c757a..3de920cf8b 100644 --- a/tests/core/test_datamodules.py +++ b/tests/core/test_datamodules.py @@ -223,6 +223,7 @@ def test_full_loop(tmpdir): default_root_dir=tmpdir, max_epochs=3, weights_summary=None, + deterministic=True, ) # fit model @@ -247,7 +248,8 @@ def test_full_loop_single_gpu(tmpdir): default_root_dir=tmpdir, max_epochs=3, weights_summary=None, - gpus=1 + gpus=1, + deterministic=True, ) # fit model @@ -273,7 +275,8 @@ def test_full_loop_dp(tmpdir): max_epochs=3, weights_summary=None, distributed_backend='dp', - gpus=2 + gpus=2, + deterministic=True, ) # fit model @@ -302,7 +305,8 @@ def test_full_loop_ddp_spawn(tmpdir): max_epochs=5, weights_summary=None, distributed_backend='ddp_spawn', - gpus=[0, 1] + gpus=[0, 1], + deterministic=True, ) # fit model