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

Reduce speed diff further, lack of GPU saturation is causing regressive times on drone CI

This commit is contained in:
SeanNaren 2020-11-27 16:28:24 +00:00
parent b4e8071de2
commit d12577d348
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
benchmarks/test_sharded_parity.py
View File

@ -63,7 +63,7 @@ def test_ddp_sharded_plugin_correctness_multi_gpu():
accelerator='ddp_spawn',
plugin=DDPShardedPlugin(),
model_cls=SeedTrainLoaderModel,
max_percent_speed_diff=0.2
max_percent_speed_diff=0.25
)
@ -79,7 +79,7 @@ def test_ddp_sharded_plugin_correctness_amp_multi_gpu():
accelerator='ddp_spawn',
plugin=DDPShardedPlugin(),
model_cls=SeedTrainLoaderModel,
max_percent_speed_diff=0.2
max_percent_speed_diff=0.25
)