From da75ac8b6cef29cf5f711c68e5059d8ae7ff3f49 Mon Sep 17 00:00:00 2001 From: lilong12 Date: Thu, 5 Dec 2019 14:52:23 +0800 Subject: [PATCH] bugfix: construct a DistributedStrategy instance if the passed one is None (#21545) --- python/paddle/fluid/incubate/fleet/collective/__init__.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python/paddle/fluid/incubate/fleet/collective/__init__.py b/python/paddle/fluid/incubate/fleet/collective/__init__.py index afb0e72eb5..481747e603 100644 --- a/python/paddle/fluid/incubate/fleet/collective/__init__.py +++ b/python/paddle/fluid/incubate/fleet/collective/__init__.py @@ -191,8 +191,10 @@ class CollectiveOptimizer(DistributedOptimizer): """ def __init__(self, optimizer, strategy=DistributedStrategy()): + if strategy is None: + strategy = DistributedStrategy() super(CollectiveOptimizer, self).__init__(optimizer, strategy) - if strategy is not None and strategy.forward_recompute: + if strategy.forward_recompute: self.forward_recompute = True self.recompute_checkpoints = strategy.recompute_checkpoints else: