From dd522a71a13788b3e65bcdb2848bd65b2ce93daa Mon Sep 17 00:00:00 2001 From: Amog Kamsetty Date: Wed, 16 Dec 2020 15:37:44 -0800 Subject: [PATCH] [SGD] Disable Elastic Training by default when using with Tune (#12927) --- python/ray/util/sgd/torch/torch_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/ray/util/sgd/torch/torch_trainer.py b/python/ray/util/sgd/torch/torch_trainer.py index d17c23191..50f20e39f 100644 --- a/python/ray/util/sgd/torch/torch_trainer.py +++ b/python/ray/util/sgd/torch/torch_trainer.py @@ -713,7 +713,7 @@ class BaseTorchTrainable(Trainable): "removed in " "a future version of Ray. Override Trainable.step instead.") - train_stats = self.trainer.train(max_retries=10, profile=True) + train_stats = self.trainer.train(max_retries=0, profile=True) validation_stats = self.trainer.validate(profile=True) stats = merge_dicts(train_stats, validation_stats) return stats