often 5 months ago
parent
commit
2afa44f253

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

@@ -122,11 +122,12 @@ class Main(object):
             ) 
             logger.info("Begin 11111111333333") 
             fleet.init(role)
-            role = fleet._role_maker
-            
-            logger.info("is_collective:", role._is_collective)
-            logger.info("is_first_worker:", role._is_first_worker())
-            logger.info("worker_index:", role._worker_index())
+            logger.info("worker_index: %s", fleet.worker_index())
+            logger.info("is_first_worker: %s", fleet.is_first_worker())
+            logger.info("worker_num: %s", fleet.worker_num())
+            logger.info("is_distributed: %s", fleet.is_distributed())
+            logger.info("mode: %s", fleet.mode)
+
         else:
             fleet.init()