瀏覽代碼

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 = [