often 5 months ago
parent
commit
abe562feda

+ 2 - 2
recommend-model-produce/src/main/python/tools/static_ps_trainer.py

@@ -113,10 +113,10 @@ class Main(object):
     def init_fleet_with_gloo(use_gloo=True):
         if use_gloo:
             os.environ["PADDLE_WITH_GLOO"] = "1"
-            role = PaddleCloudRoleMaker(
+            role = role_maker.PaddleCloudRoleMaker(
                 is_collective=False,
                 init_gloo=True
-            )
+            )            
             fleet.init(role)
         else:
             fleet.init()