often 5 maanden geleden
bovenliggende
commit
858b410399
1 gewijzigde bestanden met toevoegingen van 4 en 1 verwijderingen
  1. 4 1
      recommend-model-produce/src/main/python/tools/static_ps_trainer.py

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

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