소스 검색

add infer code

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

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

@@ -269,7 +269,7 @@ class Main(object):
         self.count_method = self.config.get("runner.example_count_method",
                                             "example")
 
-    def dataset_train_loop(self, epoch):
+    def dataset_train_loop(self, epoch=0):
         logger.info("Epoch: {}, Running Dataset Begin.".format(epoch))
 
         fetch_info = [