소스 검색

add debug info

often 5 달 전
부모
커밋
d677506f3b
1개의 변경된 파일2개의 추가작업 그리고 1개의 파일을 삭제
  1. 2 1
      recommend-model-produce/src/main/python/tools/static_ps_trainer.py

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

@@ -271,7 +271,8 @@ class Main(object):
                 "dump_fields_path": dump_fields_path,
                 "dump_fields": config.get("runner.dump_fields")
             })
-        print(paddle.static.default_main_program()._fleet_opt)
+        logger.info(paddle.static.default_main_program()._fleet_opt)
+        logger.info("Epoch: {}, debug debug debug debug.".format(epoch))
         self.exe.train_from_dataset(
             program=paddle.static.default_main_program(),
             dataset=self.reader,