From df7a52df5f021cf8617f1c07e1692f5d1b140d10 Mon Sep 17 00:00:00 2001 From: Sean Naren Date: Sun, 29 Nov 2020 19:46:51 +0000 Subject: [PATCH] Increase multiple optimizers parity for drone CI (#4884) --- benchmarks/test_sharded_parity.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmarks/test_sharded_parity.py b/benchmarks/test_sharded_parity.py index 2ecdc6c50e..e71288b827 100644 --- a/benchmarks/test_sharded_parity.py +++ b/benchmarks/test_sharded_parity.py @@ -127,7 +127,7 @@ def test_ddp_sharded_plugin_correctness_multi_gpu_multi_optim(): gpus=2, accelerator='ddp_spawn', 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 )