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

Increase multiple optimizers parity for drone CI (#4884)

This commit is contained in:
Sean Naren 2020-11-29 19:46:51 +00:00 committed by GitHub
parent 597dfa174c
commit df7a52df5f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
benchmarks/test_sharded_parity.py
View File

@ -127,7 +127,7 @@ def test_ddp_sharded_plugin_correctness_multi_gpu_multi_optim():
gpus=2, gpus=2,
accelerator='ddp_spawn', accelerator='ddp_spawn',
model_cls=SeedTrainLoaderMultipleOptimizersModel, model_cls=SeedTrainLoaderMultipleOptimizersModel,
max_percent_speed_diff=0.2 # Increase speed diff since only 2 GPUs sharding 2 optimizers max_percent_speed_diff=0.25 # Increase speed diff since only 2 GPUs sharding 2 optimizers
) )