Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

deterministic=True (#2944)

This commit is contained in:
Jirka Borovec 2020-08-13 12:29:27 +02:00 committed by GitHub
parent 2c31beccfb
commit 665c1507f0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/core/test_datamodules.py
View File

@ -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