often 5 月之前
父節點
當前提交
2afa44f253
共有 1 個文件被更改,包括 6 次插入5 次删除
  1. 6 5
      recommend-model-produce/src/main/python/tools/static_ps_trainer.py

+ 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()